diff --git a/pystencils_tests/test_version_string.py b/pystencils_tests/test_version_string.py
index 1d9c25e99c0c415072e0617bc5da830897a55db9..948515d68e56f3613ee1e4cf14e7612e76a6f36e 100644
--- a/pystencils_tests/test_version_string.py
+++ b/pystencils_tests/test_version_string.py
@@ -1,8 +1,8 @@
-import lbmpy
+import pystencils as ps
 
 
 def test_version_string():
-    version = lbmpy.__version__
+    version = ps.__version__
     print(version)
 
     numeric_version = [int(x, 10) for x in version.split('.')[0:1]]