diff --git a/onnxscript/function_libs/torch_lib/ops/nn.py b/onnxscript/function_libs/torch_lib/ops/nn.py
index de89ff6bad..ddac1644f8 100644
--- a/onnxscript/function_libs/torch_lib/ops/nn.py
+++ b/onnxscript/function_libs/torch_lib/ops/nn.py
@@ -2320,7 +2320,7 @@ def aten_unflatten_dense_tensors(
 
 
 def _get_upsample_align_corners_mode(align_corners: bool) -> str:
-    return "align_corners" if align_corners else "pytorch_half_pixel"
+    return "align_corners" if align_corners else "half_pixel"
 
 
 def _aten_upsample_output_size(
diff --git a/tests/function_libs/torch_lib/extra_opinfo.py b/tests/function_libs/torch_lib/extra_opinfo.py
index 29df92f097..7772861f14 100644
--- a/tests/function_libs/torch_lib/extra_opinfo.py
+++ b/tests/function_libs/torch_lib/extra_opinfo.py
@@ -1887,6 +1887,9 @@ def shape(size, rank, with_batch_channel=True):
     yield opinfo_core.SampleInput(make_arg(shape(D, rank)), shape(SS, rank, False), True)
 
     for align_corners in align_corners_options:
+        yield opinfo_core.SampleInput(
+            make_arg(shape(D, rank)), shape(1, rank, False), align_corners
+        )
         yield opinfo_core.SampleInput(
             make_arg(shape(D, rank)), shape(S, rank, False), align_corners
         )
@@ -1934,6 +1937,9 @@ def shape(size, rank, with_batch_channel=True):
     yield opinfo_core.SampleInput(make_arg(shape(D, rank)), shape(SS, rank, False), True, None)
 
     for align_corners in align_corners_options:
+        yield opinfo_core.SampleInput(
+            make_arg(shape(D, rank)), shape(1, rank, False), align_corners, None
+        )
         yield opinfo_core.SampleInput(
             make_arg(shape(D, rank)), shape(S, rank, False), align_corners, None
         )