diff --git a/templates/docker-compose.yml.j2 b/templates/docker-compose.yml.j2
index bdd6f40..d0c339c 100644
--- a/templates/docker-compose.yml.j2
+++ b/templates/docker-compose.yml.j2
@@ -43,9 +43,10 @@ services:
         aliases:
           - ki_db
     environment:
-      MYSQL_ROOT_PASSWORD: "{{ ki_db_root_password }}"
-      MYSQL_DATABASE: ki
-      MYSQL_USER: ki
-      MYSQL_PASSWORD: "{{ ki_db_password }}"
+      MARIADB_ROOT_PASSWORD: "{{ ki_db_root_password }}"
+      MARIADB_DATABASE: ki
+      MARIADB_USER: ki
+      MARIADB_PASSWORD: "{{ ki_db_password }}"
+      MARIADB_AUTO_UPGRADE: 1
     volumes:
       - ki_db:/var/lib/mysql/