diff --git a/rdrf/dashboards/testing.py b/rdrf/dashboards/testing.py
index 645e05e98..89e521fd7 100644
--- a/rdrf/dashboards/testing.py
+++ b/rdrf/dashboards/testing.py
@@ -56,6 +56,7 @@ def test_normal_case(self):
 
     def create_traffic_light(self, data):
         from dashboards.components.tl import TrafficLights
+        from dashboards.components.tl import get_fields
 
         vis_config = Mock()
 
@@ -95,7 +96,9 @@ def create_traffic_light(self, data):
         vis_config.config = config_dict
         patient = Mock()
 
-        return TrafficLights("crc test tl", vis_config, data, patient, None)
+        tl = TrafficLights("crc test tl", vis_config, data, patient, None)
+        tl.fields = get_fields(tl.config)
+        return tl
 
     def create_dataframe(self):
         return None