diff --git a/tests/workchains/test_aimreorworkchain.py b/tests/workchains/test_aimreorworkchain.py
index b35e181..eba80d7 100644
--- a/tests/workchains/test_aimreorworkchain.py
+++ b/tests/workchains/test_aimreorworkchain.py
@@ -5,6 +5,5 @@
 def test_setup(generate_workchain_aimreor):
     """Test `AimReorWorkChain.setup`."""
     process = generate_workchain_aimreor()
-    process.setup()
 
-    assert isinstance(process.ctx.inputs, AttributeDict)
+    assert isinstance(process.inputs, AttributeDict)