diff --git a/src/pymgrid/utils/serialize.py b/src/pymgrid/utils/serialize.py
index 45be9c2c..69d8a308 100644
--- a/src/pymgrid/utils/serialize.py
+++ b/src/pymgrid/utils/serialize.py
@@ -20,6 +20,12 @@ def add_pymgrid_yaml_representers():
         BatteryDischargeShaper
     )
 
+    from pymgrid.modules.battery.transition_models import (
+        BatteryTransitionModel,
+        BiasedTransitionModel,
+        DecayTransitionModel
+    )
+
 
 def dump_data(data_dict, stream, yaml_tag):
     if not hasattr(stream, "name"):