diff --git a/test/test_evaluate/test_gradual/test_stubs.py b/test/test_evaluate/test_gradual/test_stubs.py
index 4e58c881..63442676 100644
--- a/test/test_evaluate/test_gradual/test_stubs.py
+++ b/test/test_evaluate/test_gradual/test_stubs.py
@@ -75,5 +75,5 @@ def test_infer_and_goto(Script, code, full_name, has_stub, has_python, way,
         if only_stubs:
             assert d.is_stub()
         assert d.full_name == full_name
-        #if d.is_stub():
-            #assert d.module_path.endswith('.pyi')
+
+        assert d.is_stub() == d.module_path.endswith('.pyi')