diff --git a/frontends/pytorch/test/acap_export/test_export_cat.py b/frontends/pytorch/test/acap_export/test_export_cat.py
index f26026413517..723e4fa88736 100644
--- a/frontends/pytorch/test/acap_export/test_export_cat.py
+++ b/frontends/pytorch/test/acap_export/test_export_cat.py
@@ -42,6 +42,6 @@ def forward(self, x):
 # CHECK: "aten.convolution"
 # CHECK: torch.prim.ListConstruct
 # CHECK: "aten._cat"
-# CHECK: "aten._log_softmax"
+# CHECK: "aten._log_softmax.out"
 # CHECK: "aten.nll_loss2d_forward"
 mb.module.operation.print(large_elements_limit=2)