diff --git a/ci/test_integration.sh b/ci/test_integration.sh
index f1296acd6..573a3a9db 100755
--- a/ci/test_integration.sh
+++ b/ci/test_integration.sh
@@ -28,7 +28,7 @@ if [ "$container" == "merlin-tensorflow" ]; then
     # this vesrion of pyarrow is incompatibile
     # with the current version of cudf 22.12
     # pinning the version of pyarrow here to match the cudf-supported version
-    pip install 'feast<0.20' pyarrow==8.0.0
-    pip install dask==2022.07.1 distributed==2022.07.1
+    pip install 'feast<0.20' pyarrow==9.0.0
+    pip install dask==2022.11.1 distributed==2022.11.1
     CUDA_VISIBLE_DEVICES="$devices"  pytest -rxs tests/integration
 fi
diff --git a/tests/integration/examples/test_ci_building_deploying_multi_stage_RecSys.py b/tests/integration/examples/test_ci_building_deploying_multi_stage_RecSys.py
index 19d64c908..97c432cc2 100644
--- a/tests/integration/examples/test_ci_building_deploying_multi_stage_RecSys.py
+++ b/tests/integration/examples/test_ci_building_deploying_multi_stage_RecSys.py
@@ -79,9 +79,9 @@ def test_func():
             response = run_ensemble_on_tritonserver(
                 "/tmp/examples/poc_ensemble", ensemble.graph.input_schema, request, outputs,  "executor_model"
             )
-            response = [x.tolist()[0] for x in response["ordered_ids"]]
+            ordered_ids = [x.tolist() for x in response["ordered_ids"]]
             shutil.rmtree("/tmp/examples/", ignore_errors=True)
             """
         )
-        response = tb2.ref("response")
-        assert len(response) == top_k
\ No newline at end of file
+        ordered_ids = tb2.ref("ordered_ids")
+        assert len(ordered_ids[0]) == top_k