diff --git a/compiler_opt/tools/generate_default_trace.py b/compiler_opt/tools/generate_default_trace.py
index 240cec2..a458bc5 100644
--- a/compiler_opt/tools/generate_default_trace.py
+++ b/compiler_opt/tools/generate_default_trace.py
@@ -67,7 +67,7 @@
     except queue.Empty:
       return
     try:
-      record = runner.collect_data(module_triple, '', None)
+      record = runner.collect_data(module_triple, '', None, None)
       results_queue.put((module_triple, record))
     except:  # pylint: disable=bare-except
       logging.error('Failed to compile %s.', module_triple)