diff --git a/castellan/tests/unit/key_manager/mock_key_manager.py b/castellan/tests/unit/key_manager/mock_key_manager.py
index 4d3cb445..0297b2c3 100644
--- a/castellan/tests/unit/key_manager/mock_key_manager.py
+++ b/castellan/tests/unit/key_manager/mock_key_manager.py
@@ -67,7 +67,7 @@ class MockKeyManager(key_manager.KeyManager):
     def _generate_key(self, **kwargs):
         name = kwargs.get('name', None)
         algorithm = kwargs.get('algorithm', 'AES')
-        key_length = kwargs.get('key_length', 256)
+        key_length = kwargs.get('length', 256)
         _hex = self._generate_hex_key(key_length)
         return sym_key.SymmetricKey(
             algorithm,
diff --git a/castellan/tests/unit/key_manager/test_mock_key_manager.py b/castellan/tests/unit/key_manager/test_mock_key_manager.py
index 771f0984..f9d6428e 100644
--- a/castellan/tests/unit/key_manager/test_mock_key_manager.py
+++ b/castellan/tests/unit/key_manager/test_mock_key_manager.py
@@ -62,7 +62,7 @@ class MockKeyManagerTestCase(test_key_mgr.KeyManagerTestCase):
 
     def test_create_key_with_length(self):
         for length in [64, 128, 256]:
-            key_id = self.key_mgr.create_key(self.context, key_length=length)
+            key_id = self.key_mgr.create_key(self.context, length=length)
             key = self.key_mgr.get(self.context, key_id)
             self.assertEqual(length / 8, len(key.get_encoded()))