diff --git a/scripts/image b/scripts/image
index 0fbdd55f70..7874846314 100755
--- a/scripts/image
+++ b/scripts/image
@@ -95,7 +95,7 @@ if [ -n "${CUSTOM_VERSION}" ]; then
 fi
 
 LIBREELEC_ARCH="${DEVICE:-${PROJECT}}.${TARGET_ARCH}"
-if [ "${LIBREELEC_BUILD}" = "official" ]; then
+if [ "${OFFICIAL}" = "yes" ]; then
   TARGET_VERSION="${LIBREELEC_ARCH}-${LIBREELEC_VERSION}"
 else
   TARGET_VERSION="${LIBREELEC_ARCH}-${LIBREELEC_BUILD}-${LIBREELEC_VERSION}"