diff --git a/profiler/test/test_profiler_e2e.py b/profiler/test/test_profiler_e2e.py
index f5df82176ee..b38644c210c 100644
--- a/profiler/test/test_profiler_e2e.py
+++ b/profiler/test/test_profiler_e2e.py
@@ -52,7 +52,9 @@ def setUpClass(cls) -> None:
         # The serialized program file. This must live longer than cls.module,
         # because the C++ pybindings will have a pointer to it. But none of the
         # tests should need to touch it.
-        cls.__buffer: bytes = to_edge(export(model, inputs)).to_executorch().buffer
+        cls.__buffer: bytes = (
+            to_edge(export(model, inputs, strict=True)).to_executorch().buffer
+        )
 
         cls.module = _load_for_executorch_from_buffer(cls.__buffer)