diff --git a/ci/scripts/build.sh b/ci/scripts/build.sh
index 4152570c5a6c2..91b26bee8d383 100755
--- a/ci/scripts/build.sh
+++ b/ci/scripts/build.sh
@@ -40,9 +40,9 @@ cargo fmt --all -- --check
 echo "--- Build Rust components"
 
 if [[ "$profile" == "ci-dev" ]]; then
-    RISINGWAVE_FEATURE_FLAGS="--features rw-dynamic-link embedded-python-udf --no-default-features"
+    RISINGWAVE_FEATURE_FLAGS="--features rw-dynamic-link --no-default-features"
 else
-    RISINGWAVE_FEATURE_FLAGS="--features rw-static-link embedded-python-udf"
+    RISINGWAVE_FEATURE_FLAGS="--features rw-static-link"
 fi
 
 cargo build \
@@ -53,6 +53,7 @@ cargo build \
     -p risingwave_compaction_test \
     -p risingwave_e2e_extended_mode_test \
     $RISINGWAVE_FEATURE_FLAGS \
+    --features embedded-python-udf \
     --profile "$profile"