diff --git a/tests/test_generator.py b/tests/test_generator.py
index 4c25ccd05..0feea1e86 100644
--- a/tests/test_generator.py
+++ b/tests/test_generator.py
@@ -103,13 +103,11 @@ def files():
 def mock_job() -> unittest.mock.MagicMock:
     """Fixture to create a mock Job."""
     job = unittest.mock.MagicMock(
-        **{
-            "entity": EchoHelloWorldEntity(),
-            "name": "test_job",
-            "get_launch_steps": unittest.mock.MagicMock(
-                side_effect=lambda: NotImplementedError()
-            ),
-        },
+        get_launch_steps=unittest.mock.MagicMock(
+            side_effect=lambda: NotImplementedError()
+        ),
+        entity=EchoHelloWorldEntity(),
+        name="test_job",
         spec=Job,
     )
     yield job