diff --git a/laborious/activities/mlflow.py b/laborious/activities/mlflow.py
index b5c3d8a..4e5e3a6 100644
--- a/laborious/activities/mlflow.py
+++ b/laborious/activities/mlflow.py
@@ -95,6 +95,7 @@ class MLFlow(BaseActivity):
         response_data = self.model_monitoring_repository.predict(
             model_name, data, model_retention)
 
+        self.debug("Prediction response data:", metadata)
         self.debug(response_data, metadata)
 
         return response_data
diff --git a/laborious/utils/repository/model_repository.py b/laborious/utils/repository/model_repository.py
index edc99e8..c3fe550 100644
--- a/laborious/utils/repository/model_repository.py
+++ b/laborious/utils/repository/model_repository.py
@@ -68,7 +68,7 @@ class MLFlowRepository():
         try:
             start_time = datetime.now()
             data = self.model_serving.get_cached_predict(
-                model_name, data, model_retention)[0:1]
+                model_name, data, model_retention)
 
             end_time = datetime.now()
             data = pd.DataFrame(data, columns=['prediction'])
diff --git a/values.yaml b/values.yaml
index b3025ae..490dba6 100644
--- a/values.yaml
+++ b/values.yaml
@@ -3,7 +3,7 @@
 # Declare variables to be passed into your templates.
 
 # This will set the replicaset count more information can be found here: https://kubernetes.io/docs/concepts/workloads/controllers/replicaset/
-replicaCount: 3
+replicaCount: 1
 
 # This sets the container image more information can be found here: https://kubernetes.io/docs/concepts/containers/images/
 image: