diff --git a/.github/workflows/release.yaml b/.github/workflows/release.yaml
index 21038a8..da16452 100644
--- a/.github/workflows/release.yaml
+++ b/.github/workflows/release.yaml
@@ -10,8 +10,28 @@ env:
   MLFLOW_UI_IMAGE_NAME: cogstacksystems/cogstack-mlflow-ui
 
 jobs:
+  ensure-branch:
+    runs-on: ubuntu-latest
+    outputs:
+      is-valid: ${{ steps.ensure-branch.outputs.is-valid }}
+    steps:
+      - name: Ensures release is from the production branch only
+        id: ensure-branch
+        run: |
+          TARGET_BRANCH="${{ github.event.release.target_commitish }}"
+          if [ "$TARGET_BRANCH" != "production" ]; then
+            echo "Only releases from the 'production' branch are allowed but found: $TARGET_BRANCH"
+            echo "is-valid=false" >> "$GITHUB_OUTPUT"
+            exit 1
+          else
+            echo "Target release branch is: $TARGET_BRANCH"
+            echo "is-valid=true" >> "$GITHUB_OUTPUT"
+          fi
+
   qc:
     runs-on: ubuntu-latest
+    needs: ensure-branch
+    if: needs.ensure-branch.outputs.is-valid == 'true'
     steps:
       - uses: actions/checkout@v4
       - name: Install uv
@@ -31,7 +51,8 @@ jobs:
 
   release:
     runs-on: ubuntu-latest
-    needs: qc
+    needs: [ensure-branch, qc]
+    if: needs.ensure-branch.outputs.is-valid == 'true'
     permissions:
       contents: read
       packages: write
@@ -90,7 +111,8 @@ jobs:
 
   release-extra:
     runs-on: ubuntu-latest
-    needs: qc
+    needs: [ensure-branch, qc]
+    if: needs.ensure-branch.outputs.is-valid == 'true'
     permissions:
       contents: read
       packages: write
diff --git a/app/__init__.py b/app/__init__.py
index 3dc1f76..d3ec452 100644
--- a/app/__init__.py
+++ b/app/__init__.py
@@ -1 +1 @@
-__version__ = "0.1.0"
+__version__ = "0.2.0"
diff --git a/docker-compose-mlflow.yml b/docker-compose-mlflow.yml
index ec73462..d31a17a 100644
--- a/docker-compose-mlflow.yml
+++ b/docker-compose-mlflow.yml
@@ -65,7 +65,7 @@ services:
         condition: "service_healthy"
 
   mlflow-ui:
-    image: cogstacksystems/cogstack-mlflow-ui:0.1.0
+    image: cogstacksystems/cogstack-mlflow-ui:0.1.1
     restart: always
     networks:
       - cogstack-model-serve_cms
diff --git a/docker-compose.yml b/docker-compose.yml
index 1fc02dc..c486b3c 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -5,7 +5,7 @@ name: cms
 services:
 
   medcat-snomed:
-    image: cogstacksystems/cogstack-modelserve:0.1.0
+    image: cogstacksystems/cogstack-modelserve:0.1.1
     labels:
       - org.cogstack.model-serve=medcat_snomed
       - org.cogstack.model-name=SNOMED MedCAT model
@@ -52,7 +52,7 @@ services:
       start_period: 60s
 
   medcat-icd10:
-    image: cogstacksystems/cogstack-modelserve:0.1.0
+    image: cogstacksystems/cogstack-modelserve:0.1.1
     labels:
       - org.cogstack.model-serve=medcat_icd10
       - org.cogstack.model-name=ICD-10 MedCAT model
@@ -99,7 +99,7 @@ services:
       start_period: 60s
 
   medcat-opcs4:
-    image: cogstacksystems/cogstack-modelserve:0.1.0
+    image: cogstacksystems/cogstack-modelserve:0.1.1
     labels:
       - org.cogstack.model-serve=medcat_opcs4
       - org.cogstack.model-name=OPCS-4 MedCAT model
@@ -146,7 +146,7 @@ services:
       start_period: 60s
 
   medcat-deid:
-    image: cogstacksystems/cogstack-modelserve:0.1.0
+    image: cogstacksystems/cogstack-modelserve:0.1.1
     labels:
       - org.cogstack.model-serve=medcat_deid
       - org.cogstack.model-name=De-Identification MedCAT model
@@ -193,7 +193,7 @@ services:
       start_period: 60s
 
   medcat-umls:
-    image: cogstacksystems/cogstack-modelserve:0.1.0
+    image: cogstacksystems/cogstack-modelserve:0.1.1
     labels:
       - org.cogstack.model-serve=medcat_umls
       - org.cogstack.model-name=UMLS MedCAT model
@@ -240,7 +240,7 @@ services:
       start_period: 60s
 
   huggingface-ner:
-    image: cogstacksystems/cogstack-modelserve:0.1.0
+    image: cogstacksystems/cogstack-modelserve:0.1.1
     labels:
       - org.cogstack.model-serve=huggingface_ner
       - org.cogstack.model-name=HuggingFace NER model