diff --git a/python/src/com/jetbrains/python/psi/impl/PyFileEvaluator.java b/python/src/com/jetbrains/python/psi/impl/PyFileEvaluator.java
index 65e860356d00..2f77eae24a2c 100644
--- a/python/src/com/jetbrains/python/psi/impl/PyFileEvaluator.java
+++ b/python/src/com/jetbrains/python/psi/impl/PyFileEvaluator.java
@@ -145,4 +145,8 @@ public class PyFileEvaluator {
     }
     return Collections.emptyList();
   }
+
+  public Set<PyFile> getVisitedFiles() {
+    return myVisitedFiles;
+  }
 }