diff --git a/kamailio/defs.cfg b/kamailio/defs.cfg
index 03776c4..753828b 100644
--- a/kamailio/defs.cfg
+++ b/kamailio/defs.cfg
@@ -28,9 +28,8 @@
 #!trydef PV_BUFFER_SIZE 16384
 #!trydef PV_BUFFER_SLOTS 30
 
-#!ifndef KAMAILIO_DBMS
-#!substdef "!KAMAILIO_DBMS!kazoo!g"
-#!endif
+#!trydef KZ_DB_MODULE kazoo
+#!substdef "!KAMAILIO_DBMS!$def(KZ_DB_MODULE)!g"
 
 #!ifndef KAZOO_DATA_DIR
 #!substdef "!KAZOO_DATA_DIR!/etc/kazoo/kamailio/db!g"