diff --git a/transforms/language/text_encoder/ray/Dockerfile b/transforms/language/text_encoder/ray/Dockerfile
index 33df87171..d80910e21 100644
--- a/transforms/language/text_encoder/ray/Dockerfile
+++ b/transforms/language/text_encoder/ray/Dockerfile
@@ -5,6 +5,7 @@ FROM ${BASE_IMAGE}
 RUN pip install --no-cache-dir pytest
 
 ARG PIP_INSTALL_EXTRA_ARGS
+ARG WHEEL_FILE_NAME
 # Copy and install data processing libraries 
 # These are expected to be placed in the docker context before this is run (see the make image).
 COPY --chown=ray:users data-processing-dist data-processing-dist