diff --git a/docs-generator/main.py b/docs-generator/main.py
index c4eee9545..d7a5bbad3 100644
--- a/docs-generator/main.py
+++ b/docs-generator/main.py
@@ -18,7 +18,7 @@ def main():
     parser.add_argument(
         "--core_factory_file",
         help="Path to core operator factory",
-        default="inference/inference-core/src/commonMain/kotlin/io/kinference.core/operators/KIOperatorFactory.kt",
+        default="inference/inference-core/src/jvmMain/kotlin/io/kinference.core/operators/KIOperatorFactory.kt",
         required=False
     )