diff --git a/tests/unittest/compiler/test_slice_permute021_fusion.py b/tests/unittest/compiler/test_slice_permute021_fusion.py
index ea6d57c0c..d7e8b40c8 100644
--- a/tests/unittest/compiler/test_slice_permute021_fusion.py
+++ b/tests/unittest/compiler/test_slice_permute021_fusion.py
@@ -123,9 +123,9 @@ def test_slice_permute021_fusion(self):
         self._test_slice_permute021_fusion(
             N=2,
             K=2,
-            slice_input_shape=[120, 1211, 1200],
+            slice_input_shape=[3, 4, 120],
             slice_start_indices=[0, 0, 3],
-            slice_end_indices=[None, None, 1100],
+            slice_end_indices=[None, None, 110],
             dims=(0, 2, 1),
             test_name="slice_permute021",
             dtype="float16",
@@ -133,9 +133,9 @@ def test_slice_permute021_fusion(self):
         self._test_slice_permute021_fusion(
             N=2,
             K=2,
-            slice_input_shape=[123, 1211, 1200],
+            slice_input_shape=[3, 121, 4],
             slice_start_indices=[0, 5, 0],
-            slice_end_indices=[None, 1200, None],
+            slice_end_indices=[None, 115, None],
             dims=(0, 2, 1),
             test_name="slice_permute021",
             dtype="float16",