diff --git a/custom-nb-image/Dockerfile b/custom-nb-image/Dockerfile
index 3d694c764..7200ed925 100644
--- a/custom-nb-image/Dockerfile
+++ b/custom-nb-image/Dockerfile
@@ -23,7 +23,7 @@ RUN pip install -r requirements.txt
 RUN pip uninstall pickle5 -y
 
 # Install codeflare-sdk and other libraries
-RUN pip install codeflare-sdk==0.4.3 \
+RUN pip install codeflare-sdk==0.4.4 \
 				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 fa4fb501a..79b0815e6 100644
--- a/custom-nb-image/VERSION
+++ b/custom-nb-image/VERSION
@@ -1 +1 @@
-sdk0.4.3-v1
+v0.4.4