diff --git a/calypsogen.py b/calypsogen.py
index 07d6c09..28ca733 100644
--- a/calypsogen.py
+++ b/calypsogen.py
@@ -22,9 +22,11 @@
 bms_messages = cangen.YAMLParser().parse(open(f"{EMBEDDED_BASE_PATH}/{module_name}/can-messages/bms.yaml", "r"))
 mpu_messages = cangen.YAMLParser().parse(open(f"{EMBEDDED_BASE_PATH}/{module_name}/can-messages/mpu.yaml", "r"))
 wheel_messages = cangen.YAMLParser().parse(open(f"{EMBEDDED_BASE_PATH}/{module_name}/can-messages/wheel.yaml", "r"))
+dti_messages = cangen.YAMLParser().parse(open(f"{EMBEDDED_BASE_PATH}/{module_name}/can-messages/dti.yaml", "r"))
 
 bms_messages.msgs.extend(mpu_messages.msgs)
 bms_messages.msgs.extend(wheel_messages.msgs)
+bms_messages.msgs.extend(dti_messages.msgs)
 
 result = cangen.RustSynth().parse_messages(bms_messages.msgs)