diff --git a/ci/test_integration.sh b/ci/test_integration.sh
index 0862f2591..bb0402e2c 100755
--- a/ci/test_integration.sh
+++ b/ci/test_integration.sh
@@ -29,4 +29,5 @@ cd /systems/
 container=$1
 devices=$2
 
+pip install feast==0.31
 CUDA_VISIBLE_DEVICES="$devices" TF_GPU_ALLOCATOR=cuda_malloc_async python -m pytest -rxs tests/integration