diff --git a/api/src/server.py b/api/src/server.py
index b78f531..22e7a7e 100644
--- a/api/src/server.py
+++ b/api/src/server.py
@@ -12,9 +12,9 @@ from PIL import Image
 import io
 import torch.nn.functional as F
 
-from src.yolo.train_yolo_faces import convert_prediction
-from src.yolo.yolo_model import Yolo_model
-from src.bird_cnn.bird_cnn import Bird_CNN
+from yolo.train_yolo_faces import convert_prediction
+from yolo.yolo_model import Yolo_model
+from bird_cnn.bird_cnn import Bird_CNN
 
 import threading