diff --git a/models.py b/models.py
index 4c00d148..109b336c 100755
--- a/models.py
+++ b/models.py
@@ -201,8 +201,7 @@ class Darknet(nn.Module):
                         x = torch.cat([layer_outputs[i] for i in layer_i], 1)
                     # print(''), [print(layer_outputs[i].shape) for i in layer_i], print(x.shape)
             elif mtype == 'shortcut':
-                layer_i = int(mdef['from'])
-                x = layer_outputs[-1] + layer_outputs[layer_i]
+                x += layer_outputs[int(mdef['from'])]
             elif mtype == 'yolo':
                 x = module(x, img_size)
                 output.append(x)