diff --git a/setup.py b/setup.py
index bb4a6da..7bf6bb1 100644
--- a/setup.py
+++ b/setup.py
@@ -7,7 +7,7 @@
 
 setuptools.setup(
     name="sagemaker_studio_image_build",
-    version="0.5.0",
+    version="0.6.0",
     author="Amazon Web Services",
     description="Build Docker Images in Amazon SageMaker Studio using AWS CodeBuild",
     long_description=long_description,