diff --git a/.github/workflows/build-and-publish.yml b/.github/workflows/build-and-publish.yml
index 64b3576..def7c68 100644
--- a/.github/workflows/build-and-publish.yml
+++ b/.github/workflows/build-and-publish.yml
@@ -16,6 +16,7 @@ on:
     branches: [ main ]
 env:
   REGISTRY: ghcr.io
+  REPO_NAME_LOWERCASE: mbi-div-b
   IMAGE_NAME_1: mbi-div-b-notebook-base
   IMAGE_NAME_2: mbi-div-b-notebook-cuda
 jobs:
@@ -62,7 +63,7 @@ jobs:
         with:
           context: ${{ env.IMAGE_NAME_2 }}
           build-args: |
-            BASE_REPO=${{ github.repository_owner }}
+            BASE_REPO=${{ env.REPO_NAME_LOWERCASE }}
             REGISTRY=${{ env.REGISTRY }}
             }}
           file: ${{ env.IMAGE_NAME_2 }}/Dockerfile
diff --git a/VERSION b/VERSION
index 0ea3a94..0c62199 100644
--- a/VERSION
+++ b/VERSION
@@ -1 +1 @@
-0.2.0
+0.2.1