diff --git a/laborious/activities/mlflow.py b/laborious/activities/mlflow.py
index 129fb75..4d542c5 100644
--- a/laborious/activities/mlflow.py
+++ b/laborious/activities/mlflow.py
@@ -423,7 +423,7 @@ class MLFlow(SientiaMonitoring):
 
 
     @activity.defn(name='get_reference_data')
-    async def get_reference_data(self, input_data: dict[str, Any]) -> list[dict[Hashable, Any]] | None:
+    async def get_reference_data(self, input_data: dict[str, Any]) -> list[dict] | None:
         """
         Get reference data from the MLflow Model Registry.