diff --git a/image_classification/resnet.py b/image_classification/resnet.py
index eeed7141671b039aada5d50c4cdd451edbf6349c..ca9330e63b32faf9dfa953ef93c4f52d86e555a4 100644
--- a/image_classification/resnet.py
+++ b/image_classification/resnet.py
@@ -51,7 +51,7 @@ def bottleneck(input, ch_in, ch_out, stride):
 def layer_warp(block_func, input, ch_in, ch_out, count, stride):
     conv = block_func(input, ch_in, ch_out, stride)
     for i in range(1, count):
-        conv = block_func(conv, ch_in, ch_out, 1)
+        conv = block_func(conv, ch_out, ch_out, 1)
     return conv