diff --git a/pystencils_tests/test_version_string.py b/pystencils_tests/test_version_string.py
index a98c1ce84dfee9bd56257eb0478212e8d93014a9..1d9c25e99c0c415072e0617bc5da830897a55db9 100644
--- a/pystencils_tests/test_version_string.py
+++ b/pystencils_tests/test_version_string.py
@@ -1,10 +1,11 @@
-import pystencils as ps
+import lbmpy
+
 
 def test_version_string():
-    version = ps.__version__
+    version = lbmpy.__version__
     print(version)
 
-    numeric_version = [int(x, 10) for x in version.split('.')[0:2]]
+    numeric_version = [int(x, 10) for x in version.split('.')[0:1]]
     test_version = sum(x * (100 ** i) for i, x in enumerate(numeric_version))
 
-    assert test_version >= 200
\ No newline at end of file
+    assert test_version >= 1