diff --git a/selftest/suite_test/test_suite/hello_world.py b/selftest/suite_test/test_suite/hello_world.py
index 9f3bf4a..073d07f 100644
--- a/selftest/suite_test/test_suite/hello_world.py
+++ b/selftest/suite_test/test_suite/hello_world.py
@@ -1,4 +1,4 @@
-from osmo_gsm_tester.test import *
+from osmo_gsm_tester.testenv import *
 
 print('hello world')
 print('I am %r / %r' % (suite.name(), test.name()))
diff --git a/selftest/suite_test/test_suite/test_error.py b/selftest/suite_test/test_suite/test_error.py
index 17b05f4..c0583ff 100755
--- a/selftest/suite_test/test_suite/test_error.py
+++ b/selftest/suite_test/test_suite/test_error.py
@@ -1,4 +1,4 @@
-from osmo_gsm_tester.test import *
+from osmo_gsm_tester.testenv import *
 
 print('I am %r / %r' % (suite.name(), test.name()))
 
diff --git a/selftest/suite_test/test_suite/test_fail.py b/selftest/suite_test/test_suite/test_fail.py
index 6880c81..cbaeded 100755
--- a/selftest/suite_test/test_suite/test_fail.py
+++ b/selftest/suite_test/test_suite/test_fail.py
@@ -1,5 +1,5 @@
 #!/usr/bin/env python3
-from osmo_gsm_tester.test import *
+from osmo_gsm_tester.testenv import *
 
 print('I am %r / %r' % (suite.name(), test.name()))