diff --git a/scripts/build_layers.sh b/scripts/build_layers.sh
index 7456a38b..2d29c578 100755
--- a/scripts/build_layers.sh
+++ b/scripts/build_layers.sh
@@ -122,15 +122,17 @@ function docker_build_zip {
             PLATFORM="manylinux2014_aarch64"
         fi
         PY_TAG="cp$(echo "$1" | tr -d '.')"
-        WHEEL_FILE=$(curl -sSfL "${S3_BASE}/index-manylinux2014.html" \
-            | grep -o "ddtrace-[^\"]*${PY_TAG}[^\"]*${PLATFORM}[^\"]*\.whl" \
+        SEARCH_PATTERN="ddtrace_serverless-[^\"]*${PY_TAG}[^\"]*${PLATFORM}[^\"]*\.whl"
+        echo "Searching for wheel ${SEARCH_PATTERN}"
+        WHEEL_FILE=$(curl -sSfL "${S3_BASE}/index-serverless.html" \
+            | grep -o "$SEARCH_PATTERN" \
             | head -n 1)
         if [ -z "${WHEEL_FILE}" ]; then
             echo "No S3 wheel found for ${PY_TAG} ${PLATFORM}, using default pyproject.toml version"
         else
             curl -sSfL "${S3_BASE}/${WHEEL_FILE}" -o "${WHEEL_FILE}"
             echo "Using S3 wheel: ${WHEEL_FILE}"
-            replace_ddtrace_dep "ddtrace = { file = \"${WHEEL_FILE}\" }"
+            replace_ddtrace_dep "ddtrace_serverless = { file = \"${WHEEL_FILE}\" }"
         fi
     fi