diff --git a/babybuddy/root/app/babybuddy/babybuddy/settings/homeassistant.py b/babybuddy/root/app/babybuddy/babybuddy/settings/homeassistant.py
new file mode 100644
index 0000000..aa4c8bb
--- /dev/null
+++ b/babybuddy/root/app/babybuddy/babybuddy/settings/homeassistant.py
@@ -0,0 +1,4 @@
+from .base import *
+
+ENABLE_HOME_ASSISTANT_SUPPORT = True
+MESSAGE_STORAGE = "django.contrib.messages.storage.session.SessionStorage"
diff --git a/babybuddy/root/etc/cont-init.d/babybuddy.sh b/babybuddy/root/etc/cont-init.d/babybuddy.sh
index ff526fb..b88eb6f 100755
--- a/babybuddy/root/etc/cont-init.d/babybuddy.sh
+++ b/babybuddy/root/etc/cont-init.d/babybuddy.sh
@@ -20,7 +20,7 @@ if [ ! -f "/config/.secretkey" ]; then
         | tr -d '\n' > /config/.secretkey
 fi
 export \
-    DJANGO_SETTINGS_MODULE="babybuddy.settings.base" \
+    DJANGO_SETTINGS_MODULE="babybuddy.settings.homeassistant" \
     ALLOWED_HOSTS="${ALLOWED_HOSTS:-*}" \
     TIME_ZONE="${TZ:-UTC}" \
     DEBUG="${DEBUG:-False}" \
diff --git a/babybuddy/root/etc/services.d/babybuddy/run b/babybuddy/root/etc/services.d/babybuddy/run
index 5dfba22..3dc9af6 100755
--- a/babybuddy/root/etc/services.d/babybuddy/run
+++ b/babybuddy/root/etc/services.d/babybuddy/run
@@ -14,7 +14,7 @@ options+=(--worker-class=gthread)
 cd /app/babybuddy || bashio.exit.nok 'Failed cd, /app/babybuddy does not exist, exiting...'
 
 export \
-    DJANGO_SETTINGS_MODULE="babybuddy.settings.base" \
+    DJANGO_SETTINGS_MODULE="babybuddy.settings.homeassistant" \
     ALLOWED_HOSTS="${ALLOWED_HOSTS:-*}" \
     TIME_ZONE="${TZ:-UTC}" \
     DEBUG="${DEBUG:-False}" \