diff --git a/protobuf-lite/build.gradle b/protobuf-lite/build.gradle
index 23e798e0ee..cd43466a8c 100644
--- a/protobuf-lite/build.gradle
+++ b/protobuf-lite/build.gradle
@@ -57,3 +57,9 @@ protobuf {
     }
   }
 }
+
+idea {
+    module {
+        sourceDirs += file("${projectDir}/build/generated/source/proto/test/javalite/")
+    }
+}