diff --git a/py/core/main/services/kg_service.py b/py/core/main/services/kg_service.py
index 856f341a4..d0024ed65 100644
--- a/py/core/main/services/kg_service.py
+++ b/py/core/main/services/kg_service.py
@@ -4,7 +4,7 @@
 from uuid import UUID
 
 
-from core.base import KGCreationStatus, RunLoggingSingleton, RunManager
+from core.base import KGExtractionStatus, RunLoggingSingleton, RunManager
 from core.base.abstractions import KGCreationSettings, KGEnrichmentSettings
 
 from core.base.abstractions import GenerationConfig