diff --git a/tests/metrics/machine_learning/tensorflow_mobilenet_benchmark.sh b/tests/metrics/machine_learning/tensorflow_mobilenet_benchmark.sh
index 72950b5b42..6bc7598d83 100755
--- a/tests/metrics/machine_learning/tensorflow_mobilenet_benchmark.sh
+++ b/tests/metrics/machine_learning/tensorflow_mobilenet_benchmark.sh
@@ -10,8 +10,8 @@ set -o pipefial
 SCRIPT_PATH=$(dirname "$(readlink -f "$0")")
 source "${SCRIPT_PATH}/../lib/common.bash"
 
-IMAGE="docker.io/library/amextdx:latest"
-DOCKERFILE="${SCRIPT_PATH}/amex_tensorflow_dockerfile/Dockerfile"
+IMAGE="docker.io/library/tensorflowmobilenet:latest"
+DOCKERFILE="${SCRIPT_PATH}/tensorflow_mobilenet_dockerfile/Dockerfile"
 tensorflow_file=$(mktemp tensorflowresults.XXXXXXXXXX)
 NUM_CONTAINERS="$1"
 TIMEOUT="$2"
diff --git a/tests/metrics/machine_learning/tensorflow_mobilenet_dockerfile/Dockerfile b/tests/metrics/machine_learning/tensorflow_mobilenet_dockerfile/Dockerfile
new file mode 100644
index 0000000000..a7672bd7fe
--- /dev/null
+++ b/tests/metrics/machine_learning/tensorflow_mobilenet_dockerfile/Dockerfile
@@ -0,0 +1,21 @@
+# Copyright (c) 2023 Intel Corporation
+#
+# SPDX-License-Identifier: Apache-2.0
+
+# Usage: FROM [image name]
+FROM ubuntu:20.04
+
+ENV DEBIAN_FRONTEND=noninteractive
+
+# Version of the Dockerfile
+LABEL DOCKERFILE_VERSION="1.0"
+
+RUN apt-get update && \
+	apt-get install -y --no-install-recommends wget nano curl build-essential git && \
+	apt-get install -y python3.8  python3-pip && \
+	pip install intel-tensorflow-avx512==2.8.0 && \
+	pip install protobuf==3.20.* && \
+	wget https://storage.googleapis.com/intel-optimized-tensorflow/models/v1_8/mobilenet_v1_1.0_224_frozen.pb && \
+	git clone https://github.com/IntelAI/models.git
+
+CMD ["/bin/bash"]