diff --git a/bin/test.sh b/bin/test.sh
index 51d1c2fdcf8de..8a33cfd163fc9 100755
--- a/bin/test.sh
+++ b/bin/test.sh
@@ -36,10 +36,13 @@ build_docker() {
 
 build_universe() {
     # create universe
+    CMD="docker images ${FULL_DOCKER_IMAGE} | grep ${FULL_DOCKER_IMAGE} | awk '{print $3}'"
+    DOCKER_IMAGE_ID=$("${CMD}")
     jq --arg version ${VERSION} \
        --arg uri ${SPARK_URI} \
        --arg image ${FULL_DOCKER_IMAGE} \
-       '{python_package, "version": $version, "spark_uri": $uri, "docker_image": $image}' \
+       --arg image_id ${DOCKER_IMAGE_ID} \
+       '{python_package, "version": $version, "spark_uri": $uri, "docker_image": $image, "docker_image_id": $image_id}' \
        manifest.json > manifest.json.tmp
     mv manifest.json.tmp manifest.json
     ./bin/make-package.py