diff --git a/extension/android_test/add_model.py b/extension/android_test/add_model.py
index 5c7cf4770e2..b7ac3955ee3 100644
--- a/extension/android_test/add_model.py
+++ b/extension/android_test/add_model.py
@@ -13,7 +13,7 @@ def forward(self, x: torch.Tensor, y: torch.Tensor):
 
 
 # 1. torch.export: Defines the program with the ATen operator set.
-aten_dialect = export(Add(), (torch.ones(1), torch.ones(1)))
+aten_dialect = export(Add(), (torch.ones(1), torch.ones(1)), strict=True)
 
 # 2. to_edge: Make optimizations for Edge devices
 edge_program = to_edge(aten_dialect)