diff --git a/backend/app/database/falkor/code-graph-backend/api/analyzers/analyzer.py b/backend/app/database/falkor/code-graph-backend/api/analyzers/analyzer.py
index 24c6f89f..d805d746 100644
--- a/backend/app/database/falkor/code-graph-backend/api/analyzers/analyzer.py
+++ b/backend/app/database/falkor/code-graph-backend/api/analyzers/analyzer.py
@@ -2,8 +2,8 @@
 from typing import Optional
 
 from tree_sitter import Language, Node, Parser, Point
-from api.entities.entity import Entity
-from api.entities.file import File
+from ..entities.entity import Entity
+from ..entities.file import File
 from abc import ABC, abstractmethod
 from multilspy import SyncLanguageServer
 
diff --git a/backend/app/database/falkor/code-graph-backend/api/analyzers/source_analyzer.py b/backend/app/database/falkor/code-graph-backend/api/analyzers/source_analyzer.py
index 12502ab9..f1f000f9 100644
--- a/backend/app/database/falkor/code-graph-backend/api/analyzers/source_analyzer.py
+++ b/backend/app/database/falkor/code-graph-backend/api/analyzers/source_analyzer.py
@@ -2,8 +2,8 @@
 from pathlib import Path
 from typing import Optional
 
-from api.entities.entity import Entity
-from api.entities.file import File
+from ..entities.entity import Entity
+from ..entities.file import File
 
 from ..graph import Graph
 from .analyzer import AbstractAnalyzer
diff --git a/backend/app/database/falkor/code-graph-backend/api/entities/entity.py b/backend/app/database/falkor/code-graph-backend/api/entities/entity.py
index e2716934..cd0f17d6 100644
--- a/backend/app/database/falkor/code-graph-backend/api/entities/entity.py
+++ b/backend/app/database/falkor/code-graph-backend/api/entities/entity.py
@@ -1,4 +1,5 @@
-from typing import Callable, Self
+from typing import Callable
+from typing_extensions import Self
 from tree_sitter import Node
 
 
diff --git a/backend/app/database/falkor/code-graph-backend/api/entities/file.py b/backend/app/database/falkor/code-graph-backend/api/entities/file.py
index c59e2b6a..0833f64f 100644
--- a/backend/app/database/falkor/code-graph-backend/api/entities/file.py
+++ b/backend/app/database/falkor/code-graph-backend/api/entities/file.py
@@ -1,7 +1,7 @@
 from pathlib import Path
 from tree_sitter import Node, Tree
 
-from api.entities.entity import Entity
+from .entity import Entity
 
 
 class File:
diff --git a/backend/app/database/falkor/code-graph-backend/api/index.py b/backend/app/database/falkor/code-graph-backend/api/index.py
index 0a87dd5e..545532a1 100644
--- a/backend/app/database/falkor/code-graph-backend/api/index.py
+++ b/backend/app/database/falkor/code-graph-backend/api/index.py
@@ -6,13 +6,13 @@
 from dotenv import load_dotenv
 from flask import Flask, request, jsonify
 
-from api.analyzers.source_analyzer import SourceAnalyzer
-from api.git_utils import git_utils
-from api.git_utils.git_graph import GitGraph
-from api.graph import Graph, get_repos, graph_exists
-from api.info import get_repo_info
-from api.llm import ask
-from api.project import Project
+from .analyzers.source_analyzer import SourceAnalyzer
+from .git_utils import git_utils
+from .git_utils.git_graph import GitGraph
+from .graph import Graph, get_repos, graph_exists
+from .info import get_repo_info
+from .llm import ask
+from .project import Project
 from .auto_complete import prefix_search
 
 # Load environment variables from .env file
diff --git a/backend/app/database/falkor/code-graph-backend/tests/index.py b/backend/app/database/falkor/code-graph-backend/tests/index.py
index 63c95cc5..44b90010 100644
--- a/backend/app/database/falkor/code-graph-backend/tests/index.py
+++ b/backend/app/database/falkor/code-graph-backend/tests/index.py
@@ -4,8 +4,8 @@
 from functools import wraps
 from dotenv import load_dotenv
 
-from api.project import Project
-from api.auto_complete import prefix_search
+from ..api.project import Project
+from ..api.auto_complete import prefix_search
 from flask import Flask, request, jsonify
 
 # Load environment variables from .env file