diff --git a/custom-nb-image/Dockerfile b/custom-nb-image/Dockerfile
index 24da085ba..16ad45189 100644
--- a/custom-nb-image/Dockerfile
+++ b/custom-nb-image/Dockerfile
@@ -22,8 +22,8 @@ RUN pip install -r requirements.txt
 
 RUN pip uninstall pickle5 -y 
 
-# Install codeflare-cli and other libraries
-RUN pip install codeflare-sdk==0.2.1 \
+# Install codeflare-sdk and other libraries
+RUN pip install codeflare-sdk==0.2.2 \
 				datasets==2.6.1 \
 				transformers==4.23.1 \
 				evaluate==0.3.0 \
diff --git a/custom-nb-image/VERSION b/custom-nb-image/VERSION
index b1803f36c..0975e046a 100644
--- a/custom-nb-image/VERSION
+++ b/custom-nb-image/VERSION
@@ -1 +1 @@
-sdk0.2.1-v2
+sdk0.2.2-v1
diff --git a/custom-nb-image/requirements.txt b/custom-nb-image/requirements.txt
index 44938879e..17822fabf 100644
--- a/custom-nb-image/requirements.txt
+++ b/custom-nb-image/requirements.txt
@@ -157,7 +157,7 @@ python-json-logger==2.0.4; python_version >= '3.5'
 pytz==2022.2.1
 pyyaml==6.0; python_version >= '3.6'
 pyzmq==24.0.1; python_version >= '3.6'
-ray==2.1.0
+ray[default]==2.1.0
 requests-oauthlib==1.3.1; python_version >= '2.7' and python_version not in '3.0, 3.1, 3.2, 3.3'
 requests==2.28.1; python_version >= '3.7' and python_version < '4'
 rsa==4.9; python_version >= '3.6'
diff --git a/pyproject.toml b/pyproject.toml
index 60d23c427..408a49d9d 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "codeflare-sdk"
-version = "0.2.1"
+version = "0.2.2"
 description = "Python SDK for codeflare client"
 
 license = "Apache-2.0"
@@ -20,7 +20,7 @@ homepage = "https://github.com/project-codeflare/codeflare-sdk"
 keywords = ['codeflare', 'python', 'sdk', 'client', 'batch', 'scale']
 
 [tool.poetry.dependencies]
-python = "^3.6"
+python = "^3.6.3"
 openshift-client = "1.0.18"
 rich = "^12.5"
-ray = "2.1.0"
+ray = {version = "2.1.0", extras = ["default"]}