diff --git a/backends/arm/test/misc/test_dim_order.py b/backends/arm/test/misc/test_dim_order.py
index 862f005f34e..14e12461652 100644
--- a/backends/arm/test/misc/test_dim_order.py
+++ b/backends/arm/test/misc/test_dim_order.py
@@ -17,7 +17,7 @@
 )
 
 
-input_t1 = Tuple[torch.Tensor]  # Input x
+input_t1 = Tuple[torch.Tensor, ...]  # Input x
 
 
 class ChannelsLastInput(torch.nn.Module):