diff --git a/tests/nuodb_executionflow_test.py b/tests/nuodb_executionflow_test.py
index 1701ae6..2e92afd 100755
--- a/tests/nuodb_executionflow_test.py
+++ b/tests/nuodb_executionflow_test.py
@@ -9,7 +9,7 @@
 
 import unittest
 import os
-from distutils.version import StrictVersion
+from distutils.version import LooseVersion
 
 from .nuodb_base import NuoBase
 from pynuodb.exception import Error
@@ -194,7 +194,7 @@ def test_error_after_error(self):
 
 def version_lt(version):
     current_version = os.getenv('NUODB_VERSION', None)
-    if current_version is not None and StrictVersion(current_version) < StrictVersion(version):
+    if current_version is not None and LooseVersion(current_version) < LooseVersion(version):
         return True
 
     return False