diff --git a/source/python/gui/source/tests/test_graphs.py b/source/python/gui/source/tests/test_graphs.py
index bc232fa9..2ef63074 100644
--- a/source/python/gui/source/tests/test_graphs.py
+++ b/source/python/gui/source/tests/test_graphs.py
@@ -66,6 +66,7 @@
 ]
 
 samples_df_expected_locations = [
+    "main",
     "/home/jose/omnitrace/examples/causal/causal.cpp:103",
     "/home/jose/omnitrace/examples/causal/causal.cpp:110",
     "/home/jose/omnitrace/examples/causal/causal.cpp:112",