diff --git a/laborious/activities/model_metrics.py b/laborious/activities/model_metrics.py
index a3deda8..dce8523 100644
--- a/laborious/activities/model_metrics.py
+++ b/laborious/activities/model_metrics.py
@@ -1,7 +1,7 @@
 from temporalio import activity, workflow
 
 with workflow.unsafe.imports_passed_through():
-    from typing import Any, Hashable
+    from typing import Any
 
     from pandas import DataFrame, Index, to_datetime
 
@@ -138,7 +138,7 @@ class ModelMetrics(SientiaMonitoring):
 
 
     @activity.defn(name='calculate_drift')
-    async def calculate_drift(self, input_data: dict[str, Any]) -> list:
+    async def calculate_drift(self, input_data: dict[str, Any]) -> list[dict]:
         """
         Calculate drift metrics for a model.
 
@@ -266,7 +266,7 @@ class ModelMetrics(SientiaMonitoring):
         return drift_df.to_dict(orient='records')
 
     @activity.defn(name='calculate_simple_metrics')
-    async def calculate_simple_metrics(self, input_data: dict[str, Any]) -> list[dict[Hashable, Any]]:
+    async def calculate_simple_metrics(self, input_data: dict[str, Any]) -> list[dict]:
         """
         Calculate simple metrics for a model. Metrics available are:
             - rmse