diff --git a/experiments_woz/mt_gauss.py b/experiments_woz/mt_gauss.py
index 268ac02a39e8fa824c8b0fc8f4225a48ff386f90..2e8daa35169649fbe3be6a118ce3b02275289a5a 100644
--- a/experiments_woz/mt_gauss.py
+++ b/experiments_woz/mt_gauss.py
@@ -68,7 +68,7 @@ def main(seed):
         avg_type='word',
         # avg_type='slot',
         # slot_weight=10,
-        use_aux_kl=False,
+        use_aux_kl=True,
         print_step=300,
         ckpt_step=1416,
         improve_threshold=0.996,