diff --git a/tools/check_package_release.py b/tools/check_package_release.py
index a501080ff..4009f5c56 100644
--- a/tools/check_package_release.py
+++ b/tools/check_package_release.py
@@ -5,7 +5,7 @@
 
 import requests
 
-AVAILABLE_PACKAGES = ["vizro-core"]
+AVAILABLE_PACKAGES = ["vizro-core","vizro-ai"]
 VERSION_MATCHSTR = r'\s*__version__\s*=\s*"(\d+\.\d+\.\d+)"'
 RESPONSE_ERROR = 404
 
diff --git a/tools/release.sh b/tools/release.sh
index 1e58b4420..c14929beb 100755
--- a/tools/release.sh
+++ b/tools/release.sh
@@ -13,9 +13,15 @@ GITHUB_ENDPOINT="https://api.github.com/repos/${GITHUB_USER}/${GITHUB_REPO}/rele
 
 BODY=$(jq -Rs . < "$BODY_PATH")
 
+if [ "$PACKAGE" = "vizro-core" ]; then
+  TAG_NAME="${VERSION}"
+else
+  TAG_NAME="${PACKAGE}-${VERSION}"
+fi
+
 PAYLOAD=$(cat <<-END
 {
-    "tag_name": "${VERSION}",
+    "tag_name": "${TAG_NAME}",
     "target_commitish": "main",
     "name": "${PACKAGE}-${VERSION}",
     "body": ${BODY},