diff --git a/testing/codeflare/images/Containerfile.codeflare_sdk b/testing/codeflare/images/Containerfile.codeflare_sdk
index 60be0f860..22e9ccc9b 100644
--- a/testing/codeflare/images/Containerfile.codeflare_sdk
+++ b/testing/codeflare/images/Containerfile.codeflare_sdk
@@ -2,4 +2,5 @@
 FROM ci-artifacts:main
 
 RUN pip install --no-cache-dir  \
-           codeflare_sdk
+           -e git+https://github.com/project-codeflare/codeflare-sdk.git@refs/pull/254/head#egg=codeflare-sdk
+RUN pip install -r https://raw.githubusercontent.com/project-codeflare/codeflare-sdk/main/requirements.txt
diff --git a/testing/codeflare/sdk_user/sample/sample.py b/testing/codeflare/sdk_user/sample/sample.py
index 72a2951c7..096bae80a 100644
--- a/testing/codeflare/sdk_user/sample/sample.py
+++ b/testing/codeflare/sdk_user/sample/sample.py
@@ -35,16 +35,7 @@ def main():
 
     # Bring up the cluster
     cluster.up()
-    while True:
-        try:
-            cluster.wait_ready()
-            break
-        except TypeError as e:
-            # there's a timeout in the calling script
-            logging.warning(f"cluster.wait_ready() failed with {e}")
-            import time
-            time.sleep(5)
-
+    cluster.wait_ready()
     cluster.status()
     cluster.details()