diff --git a/services/ml/app/main.py b/services/ml/app/main.py
index 71000fa..65b0f67 100644
--- a/services/ml/app/main.py
+++ b/services/ml/app/main.py
@@ -13,6 +13,7 @@ import json
 from fastapi import FastAPI, HTTPException, status
 from fastapi.middleware.cors import CORSMiddleware
 from pydantic import BaseModel, Field
+import numpy as np
 import pandas as pd
 import joblib
 import mlflow