diff --git a/tests/entrypoint.sh b/tests/entrypoint.sh
index eefd4b8..49f88f4 100755
--- a/tests/entrypoint.sh
+++ b/tests/entrypoint.sh
@@ -3,11 +3,11 @@ set -euxo pipefail
 
 listen_address=
 if [[ ${TEST_LOCAL_CODEGEN:-false} == "true" ]]; then
-      if [[ ${CI:-""} != "" ]]; then
+      if [[ -z $CI ]]; then
+        listen_address="http://host.docker.internal:9000"
+      else
          # Codegen address when running test
          listen_address="http://172.17.0.1:9000"
-      else
-         listen_address="http://host.docker.internal:9000"
       fi
 else
     listen_address="https://codegen-staging.substreams.dev"