diff --git a/java/src/processing/mode/java/JavaEditor.java b/java/src/processing/mode/java/JavaEditor.java
index 2c6878815..c8e8a57c1 100644
--- a/java/src/processing/mode/java/JavaEditor.java
+++ b/java/src/processing/mode/java/JavaEditor.java
@@ -142,9 +142,11 @@ public class JavaEditor extends Editor {
 
     { // Init error checker
       errorCheckerService = new ErrorCheckerService(this);
-      Document currentDocument = currentDocument();
-      if (currentDocument != null) {
-        errorCheckerService.addListener(currentDocument);
+      for (SketchCode code : getSketch().getCode()) {
+        Document document = code.getDocument();
+        if (document != null) {
+          errorCheckerService.addListener(document);
+        }
       }
       errorCheckerService.start();
       errorCheckerService.request();