diff --git a/transactron/testing/testbenchio.py b/transactron/testing/testbenchio.py
index 7611a1e..9af6340 100644
--- a/transactron/testing/testbenchio.py
+++ b/transactron/testing/testbenchio.py
@@ -127,6 +127,7 @@ def handle_validate_arguments():
         ret_out = mock_def_helper(self, function, arg)
         yield from self.method_return(ret_out or {})
         yield Tick()
+        yield from self.set_enable(False)
 
     def method_handle_loop(
         self,
@@ -139,7 +140,10 @@ def method_handle_loop(
         yield Passive()
         while True:
             yield from self.method_handle(
-                function, enable=enable, validate_arguments=validate_arguments, extra_settle_count=extra_settle_count
+                function,
+                enable=enable,
+                validate_arguments=validate_arguments,
+                extra_settle_count=extra_settle_count,
             )
 
     # Debug signals