diff --git a/src/engine/src/helpers/TemplateMapper.py b/src/engine/src/helpers/TemplateMapper.py
index c2dbee97..5a8819d7 100644
--- a/src/engine/src/helpers/TemplateMapper.py
+++ b/src/engine/src/helpers/TemplateMapper.py
@@ -1,11 +1,10 @@
 from typing import Union
 
-from .TemplateRepository import TemplateRepository
-from ..owe_python_sdk.schema import (
+from TemplateRepository import TemplateRepository
+from owe_python_sdk.schema import (
     Uses,
     Pipeline,
     Task,
-    TemplateTask,
     FunctionTask,
     ApplicationTask,
     ImageBuildTask,
diff --git a/src/engine/src/helpers/TemplateRepository.py b/src/engine/src/helpers/TemplateRepository.py
index 2964e8f9..33c32fa3 100644
--- a/src/engine/src/helpers/TemplateRepository.py
+++ b/src/engine/src/helpers/TemplateRepository.py
@@ -1,7 +1,7 @@
 import os,json
 
-from .GitCacheService import GitCacheService
-from ..owe_python_sdk.schema import Uses
+from GitCacheService import GitCacheService
+from owe_python_sdk.schema import Uses
 
 class TemplateRepository:
     def __init__(self, cache_dir: str):