diff --git a/src/india_api/internal/inputs/indiadb/client.py b/src/india_api/internal/inputs/indiadb/client.py
index 966f528..aafd954 100644
--- a/src/india_api/internal/inputs/indiadb/client.py
+++ b/src/india_api/internal/inputs/indiadb/client.py
@@ -106,7 +106,7 @@ def get_predicted_power_production_for_location(
 
             if site.ml_model is not None:
                 ml_model_name = site.ml_model.model_name
-            log.info(f"Using ml model {site.ml_model.model_name}")
+            log.info(f"Using ml model {ml_model_name}")
 
             # read actual generations
             values = get_latest_forecast_values_by_site(
@@ -295,7 +295,7 @@ def get_site_forecast(self, site_uuid: str, email: str) -> list[internal.Predict
             site = get_site_by_uuid(session=session, site_uuid=site_uuid)
             if site.ml_model is not None:
                 ml_model_name = site.ml_model.model_name
-            log.info(f"Using ml model {site.ml_model.model_name}")
+            log.info(f"Using ml model {ml_model_name}")
 
             if isinstance(site_uuid, str):
                 site_uuid = UUID(site_uuid)