diff --git a/MaxText/tests/profiler_test.py b/MaxText/tests/profiler_test.py
index 1c2cf3780e..c4ef6b59ef 100644
--- a/MaxText/tests/profiler_test.py
+++ b/MaxText/tests/profiler_test.py
@@ -66,8 +66,8 @@ def test_device_trace_contains_threads(self):
       if "name" in event and event["name"] == "thread_name":
         thread_names.append((event["args"]["name"]))
     expected_threads = [
-        "TensorFlow Name Scope",
-        "TensorFlow Ops",
+        "Framework Name Scope",
+        "Framework Ops",
         "XLA Modules",
         "XLA Ops",
         "XLA TraceMe",