diff --git a/py/torch_tensorrt/_Input.py b/py/torch_tensorrt/_Input.py
index 7f36e927f0..3995eca1dd 100644
--- a/py/torch_tensorrt/_Input.py
+++ b/py/torch_tensorrt/_Input.py
@@ -265,7 +265,7 @@ def _parse_format(format: Any) -> _enums.TensorFormat:
                 return _enums.TensorFormat.channels_last
             else:
                 raise ValueError(
-                    "Provided an unsupported tensor format (support: NHCW/contiguous_format, NHWC/channel_last)"
+                    "Provided an unsupported tensor format (support: NCHW/contiguous_format, NHWC/channel_last)"
                 )
 
         elif isinstance(format, _enums.TensorFormat):