diff --git a/laborious/activities/gates.py b/laborious/activities/gates.py
index 2ac33f1..f3f7ad9 100644
--- a/laborious/activities/gates.py
+++ b/laborious/activities/gates.py
@@ -405,7 +405,7 @@ class Gates(SientiaMonitoring):
 
 
     @activity.defn(name='format_transformed_data')
-    async def format_transformed_data(self, input_data: dict[str, Any]) -> dict[str, Any]:
+    async def format_transformed_data(self, input_data: dict[str, Any]) -> dict:
         """
         Format transformed data according to configured storage policies.
         """
@@ -426,7 +426,7 @@ class Gates(SientiaMonitoring):
         return data.to_dict()
 
     @activity.defn(name='format_prediction')
-    async def format_prediction(self, input_data: dict[str, Any]) -> dict[str, Any]:
+    async def format_prediction(self, input_data: dict[str, Any]) -> dict:
         """
         Format prediction data according to configured storage policies.
 
@@ -499,7 +499,7 @@ class Gates(SientiaMonitoring):
         return data.to_dict()
 
     @activity.defn(name='format_default_prediction')
-    async def format_default_prediction(self, input_data: dict[str, Any]) -> dict[str, Any]:
+    async def format_default_prediction(self, input_data: dict[str, Any]) -> dict:
         """
         Create and format default prediction data for error conditions.
 
@@ -544,7 +544,7 @@ class Gates(SientiaMonitoring):
         return data.to_dict()
 
     @activity.defn(name='format_retrain_report')
-    async def format_retrain_report(self, input_data: dict[str, Any]) -> dict[str, Any]:
+    async def format_retrain_report(self, input_data: dict[str, Any]) -> dict:
         """
         Format retrain report data according to configured storage policies.
         """
diff --git a/requirements-light.txt b/requirements-light.txt
index 26f2290..8831822 100644
--- a/requirements-light.txt
+++ b/requirements-light.txt
@@ -3,7 +3,7 @@ psycopg2-binary
 sqlalchemy
 asyncua
 redis
-git+ssh://git@github.com/Aignosi/sientia-dataops-library.git@1.5.4
+git+ssh://git@github.com/Aignosi/sientia-dataops-library.git@1.6.1
 prometheus-client
 botocore
 boto3
diff --git a/requirements.txt b/requirements.txt
index 214e910..d475d8a 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -3,7 +3,7 @@ psycopg2-binary
 sqlalchemy
 asyncua
 redis
-git+ssh://git@github.com/Aignosi/sientia-dataops-library.git@1.5.4
+git+ssh://git@github.com/Aignosi/sientia-dataops-library.git@1.6.1
 git+ssh://git@github.com/Aignosi/sientia-mlops-library.git@0.40.2
 prometheus-client
 botocore