diff --git a/test/test_api/test_api_classes_follow_definition.py b/test/test_api/test_api_classes_follow_definition.py
index 1b6cf793..d022c69c 100644
--- a/test/test_api/test_api_classes_follow_definition.py
+++ b/test/test_api/test_api_classes_follow_definition.py
@@ -34,10 +34,7 @@ def test_follow_import_incomplete(Script, environment):
 
     # incomplete `from * import` part
     datetime = check_follow_definition_types(Script, "from datetime import datetim")
-    if environment.version_info.major == 2:
-        assert datetime == ['class']
-    else:
-        assert set(datetime) == {'class', 'instance'}  # py3: builtin and pure py version
+    assert datetime == ['class']
     # os.path check
     ospath = check_follow_definition_types(Script, "from os.path import abspat")
     assert ospath == ['function']