diff --git a/Dockerfile.QA b/Dockerfile.QA
index 1287a9d877..2c43f735a5 100644
--- a/Dockerfile.QA
+++ b/Dockerfile.QA
@@ -148,7 +148,9 @@ RUN mkdir -p qa/common && \
     cp bin/request_cancellation_test qa/L0_request_cancellation/. && \
     cp bin/triton_json_test qa/L0_json/. && \
     cp bin/backend_output_detail_test qa/L0_backend_output_detail/. && \
-    cp -r deploy/mlflow-triton-plugin qa/L0_mlflow/.
+    cp -r deploy/mlflow-triton-plugin qa/L0_mlflow/. && \
+    cp bin/input_byte_size_test qa/L0_input_validation/. && \
+    cp -r docs/examples/model_repository/simple_identity qa/L0_input_validation/models
 
 RUN mkdir -p qa/pkgs && \
     cp python/triton*.whl qa/pkgs/. && \
diff --git a/qa/L0_input_validation/test.sh b/qa/L0_input_validation/test.sh
index 47a379c52b..be7054895c 100755
--- a/qa/L0_input_validation/test.sh
+++ b/qa/L0_input_validation/test.sh
@@ -46,9 +46,10 @@ DATADIR=/data/inferenceserver/${REPO_VERSION}
 SERVER=/opt/tritonserver/bin/tritonserver
 CLIENT_LOG="./input_validation_client.log"
 TEST_PY=./input_validation_test.py
-SHAPE_TEST_PY=./input_shape_validation_test.py
 TEST_RESULT_FILE='./test_results.txt'
 SERVER_LOG="./inference_server.log"
+TEST_LOG="./input_byte_size_test.log"
+TEST_EXEC=./input_byte_size_test
 
 export CUDA_VISIBLE_DEVICES=0
 
@@ -143,6 +144,15 @@ set -e
 kill $SERVER_PID
 wait $SERVER_PID
 
+# input_byte_size_test
+set +e
+LD_LIBRARY_PATH=/opt/tritonserver/lib:$LD_LIBRARY_PATH $TEST_EXEC >>$TEST_LOG 2>&1
+if [ $? -ne 0 ]; then
+    echo -e "\n***\n*** Query Unit Test Failed\n***"
+    RET=1
+fi
+set -e
+
 if [ $RET -eq 0 ]; then
     echo -e "\n***\n*** Input Validation Test Passed\n***"
 else