diff --git a/tests/function_libs/torch_lib/extra_opinfo.py b/tests/function_libs/torch_lib/extra_opinfo.py
index 6c6b39d28..b6ccf89b1 100644
--- a/tests/function_libs/torch_lib/extra_opinfo.py
+++ b/tests/function_libs/torch_lib/extra_opinfo.py
@@ -1855,12 +1855,12 @@ def shape(size, rank, with_batch_channel=True):
     )
     yield opinfo_core.SampleInput(
         make_arg(shape(D, rank)),
-        shape(L, rank, False),
+        None,
         [1.7, 1.5, 2.0],  # scaler
     )
     yield opinfo_core.SampleInput(
         make_arg(shape(D, rank)),
-        shape(L, rank, False),
+        None,
         [0.6, 0.3, 0.5],
     )