diff --git a/smdebug/profiler/analysis/utils/merge_timelines.py b/smdebug/profiler/analysis/utils/merge_timelines.py
index 34f00b1ac..5390796d3 100644
--- a/smdebug/profiler/analysis/utils/merge_timelines.py
+++ b/smdebug/profiler/analysis/utils/merge_timelines.py
@@ -68,6 +68,8 @@ def __init__(self, path, file_suffix_filter=None, output_directory=None):
             "Forward": 10000,
             "Backward": 20000,
             "DataIterator": 30000,
+            "DataLoaderIter": 30000,
+            "DataLoaderWorker": 30000,
             "Dataset": 40000,
             "/device": 50000,
             "/host": 60000,