chore: fix problem with test

Author: darklight3it

Dockerfile.rie

@@ -22,13 +22,15 @@ RUN chmod +x build-examples.sh && HANDLERS_TO_BUILD=${HANDLERS_TO_BUILD} OUTPUT_
 # Final Image
 FROM public.ecr.aws/lambda/provided:al2023
 
+ARG OUTPUT_DIR="/var/task"
+
 ADD https://github.com/aws/aws-lambda-runtime-interface-emulator/releases/latest/download/aws-lambda-rie /usr/local/bin/aws-lambda-rie
 RUN chmod +x /usr/local/bin/aws-lambda-rie
 
 COPY scripts/custom-lambda-entrypoint.sh /usr/local/bin/lambda-entrypoint
 RUN chmod +x /usr/local/bin/lambda-entrypoint
 
-COPY --from=builder /var/task /var/task
+COPY --from=builder ${OUTPUT_DIR} /var/task
 
 ENTRYPOINT ["/usr/local/bin/lambda-entrypoint"]
 CMD ["basic-lambda"]