diff --git a/src/pricecypher/contracts/handlers/train_models_handler.py b/src/pricecypher/contracts/handlers/train_models_handler.py
index 8488876..2b3ea75 100644
--- a/src/pricecypher/contracts/handlers/train_models_handler.py
+++ b/src/pricecypher/contracts/handlers/train_models_handler.py
@@ -6,7 +6,7 @@
 
 from .base_handler import BaseHandler
 from pricecypher.enums import AccessTokenGrantType
-from pricecypher.contracts import PricecypherModel
+from ..pricecypher_model import PricecypherModel
 
 
 class TrainModelHandler(BaseHandler):