diff --git a/mealie/config.json b/mealie/config.json
index 9ea8587a3..e3c608508 100644
--- a/mealie/config.json
+++ b/mealie/config.json
@@ -110,6 +110,6 @@
   "slug": "mealie",
   "udev": true,
   "url": "https://github.com/alexbelgium/hassio-addons",
-  "version": "v2.0-beta2",
+  "version": "v2.0-beta3",
   "webui": "[PROTO:ssl]://[HOST]:[PORT:9001]"
 }
diff --git a/mealie/rootfs/run.txt b/mealie/rootfs/run.txt
index 65dffae20..74657917c 100644
--- a/mealie/rootfs/run.txt
+++ b/mealie/rootfs/run.txt
@@ -39,12 +39,19 @@ if [ -d /homeassistant/addons_config/mealie_data ] && [ ! -f /homeassistant/addo
     bashio::log.warning "Migrating data, current data will not be touched"
     cp -rnf /homeassistant/addons_config/mealie_data/* /config/ || true
     touch /homeassistant/addons_config/mealie_data/migrated
+    bashio::addon.option "DATA_DIR" "/config"
 fi
 
 if [ -f /homeassistant/addons_config/mealie/config.yaml ] && [ ! -f /homeassistant/addons_config/mealie/migrated ]; then
     bashio::log.warning "Migrating config.yaml, current data will not be touched"
     cp -nf /homeassistant/addons_config/mealie/config.yaml /config/ || true
     touch /homeassistant/addons_config/mealie/migrated
+    bashio::addon.option "DATA_DIR" "/config"
+fi
+
+if [[ "$(bashio::config "DATA_DIR")" == "/config/addons_config/mealie_data" ]]; then
+    bashio::addon.option "DATA_DIR" "/config"
+    bashio::addon.restart
 fi
 
 ###############