diff --git a/laborious/activities/model_metrics.py b/laborious/activities/model_metrics.py
index 78e7aad..a3deda8 100644
--- a/laborious/activities/model_metrics.py
+++ b/laborious/activities/model_metrics.py
@@ -138,7 +138,7 @@ class ModelMetrics(SientiaMonitoring):
 
 
     @activity.defn(name='calculate_drift')
-    async def calculate_drift(self, input_data: dict[str, Any]) -> list[dict[Hashable, Any]]:
+    async def calculate_drift(self, input_data: dict[str, Any]) -> list:
         """
         Calculate drift metrics for a model.