diff --git a/python/test/test_onnx.py b/python/test/test_onnx.py
index 68abe5fed..19b9d5f1f 100755
--- a/python/test/test_onnx.py
+++ b/python/test/test_onnx.py
@@ -24,7 +24,7 @@
 import multiprocessing
 
 BM1684X_Failed_Cases = [
-    "PadAvgPool2d", "PadEdge", "QDQ", "QDQConv", "TopK", "TorchArgMax", "TorchActivation",
+    "PadAvgPool2d", "PadEdge", "QDQ", "QDQConv", "TopK", "TorchArgmax", "TorchActivation",
     "TorchChannelShuffle"
 ]
 CV18XX_Failed_Cases = [
diff --git a/python/utils/mlir_parser.py b/python/utils/mlir_parser.py
index 063bd3f40..d77b55e43 100755
--- a/python/utils/mlir_parser.py
+++ b/python/utils/mlir_parser.py
@@ -111,8 +111,12 @@ def loc(op):
 
     @staticmethod
     def shape(op):
-        shape_type = mlir.ir.ShapedType(op.results[0].type)
-        shape = [shape_type.get_dim_size(i) for i in range(shape_type.rank)]
+        shape = []
+        for result in op.results:
+            if str(result.type) != 'none':
+                shape_type = mlir.ir.ShapedType(result.type)
+                shape = [shape_type.get_dim_size(i) for i in range(shape_type.rank)]
+                break
         return shape
 
     @staticmethod