diff --git a/src/osmo_gsm_tester/suite.py b/src/osmo_gsm_tester/suite.py
index 2313ae2..0ffa434 100644
--- a/src/osmo_gsm_tester/suite.py
+++ b/src/osmo_gsm_tester/suite.py
@@ -93,7 +93,7 @@
     def run(self, suite_run):
         assert self.suite is suite_run.definition
         with self:
-            test.setup(suite_run, self, ofono_client)
+            test.setup(suite_run, self, ofono_client, sys.modules[__name__])
             success = False
             try:
                 self.log('START')