diff --git a/src/main.py b/src/main.py
index f250eab..05a2371 100644
--- a/src/main.py
+++ b/src/main.py
@@ -16,8 +16,8 @@
 from fastapi import FastAPI, Response
 from pydantic import BaseModel
 
-from src.data.image import prepare_scenes
-from src.inference.pipeline import detect_vessels
+from data.image import prepare_scenes
+from inference.pipeline import detect_vessels
 
 app = FastAPI()