diff --git a/test/test_artifacts/v1/amazon-codewhisperer-jupyterlab-ext.test.Dockerfile b/test/test_artifacts/v2/amazon-q-developer-jupyterlab-ext.test.Dockerfile
similarity index 60%
rename from test/test_artifacts/v1/amazon-codewhisperer-jupyterlab-ext.test.Dockerfile
rename to test/test_artifacts/v2/amazon-q-developer-jupyterlab-ext.test.Dockerfile
index 1ef68a71..02599032 100644
--- a/test/test_artifacts/v1/amazon-codewhisperer-jupyterlab-ext.test.Dockerfile
+++ b/test/test_artifacts/v2/amazon-q-developer-jupyterlab-ext.test.Dockerfile
@@ -3,4 +3,4 @@ FROM $SAGEMAKER_DISTRIBUTION_IMAGE
 
 ARG MAMBA_DOCKERFILE_ACTIVATE=1
 
-CMD ["python", "-c", "import amazon_codewhisperer_jupyterlab_ext"]
+CMD ["python", "-c", "import amazon_q_developer_jupyterlab_ext"]
diff --git a/test/test_artifacts/v2/amazon-sagemaker-jupyter-ai-q-developer.test.Dockerfile b/test/test_artifacts/v2/amazon-sagemaker-jupyter-ai-q-developer.test.Dockerfile
new file mode 100644
index 00000000..4c9ec976
--- /dev/null
+++ b/test/test_artifacts/v2/amazon-sagemaker-jupyter-ai-q-developer.test.Dockerfile
@@ -0,0 +1,6 @@
+ARG SAGEMAKER_DISTRIBUTION_IMAGE
+FROM $SAGEMAKER_DISTRIBUTION_IMAGE
+
+ARG MAMBA_DOCKERFILE_ACTIVATE=1
+
+CMD ["python", "-c", "import amazon_sagemaker_jupyter_ai_q_developer"]