diff --git a/templates/env.j2 b/templates/env.j2
index 7fa128c..2a97db8 100644
--- a/templates/env.j2
+++ b/templates/env.j2
@@ -4,10 +4,14 @@
 REDIS_HOST={{ redis_host }}
 REDIS_PORT={{ redis_port }}
 # You may set DATABASE_URL instead for more advanced options
+{% if mastodon_db_login_unix_socket %}
+DB_HOST={{ mastodon_db_login_unix_socket }}
+{% else %}
 DB_HOST={{ db_host }}
-DB_USER={{ db_user }}
-DB_NAME={{ db_name }}
-DB_PASS={{ db_pass }}
+{% endif %}
+DB_USER={{ mastodon_db_user }}
+DB_NAME={{ mastodon_db }}
+DB_PASS={{ mastodon_db_password }}
 DB_PORT={{ db_port }}
 
 # Federation