diff --git a/segmentation_models_pytorch/decoders/upernet/decoder.py b/segmentation_models_pytorch/decoders/upernet/decoder.py
index 092de36a..ebcb3d10 100644
--- a/segmentation_models_pytorch/decoders/upernet/decoder.py
+++ b/segmentation_models_pytorch/decoders/upernet/decoder.py
@@ -110,7 +110,7 @@ def __init__(
             use_batchnorm=True,
         )
 
-    def forward(self, *features):
+    def forward(self, features):
         output_size = features[0].shape[2:]
         target_size = [size // 4 for size in output_size]