diff --git a/chapter05/resnet/resnet_cifar.py b/chapter05/resnet/resnet_cifar.py
index fbb0f9c2c590bc94f2783d04045e81adf38fdf8c..ad09e8cdda31bf07e614b799e36ca99ed5de128d 100644
--- a/chapter05/resnet/resnet_cifar.py
+++ b/chapter05/resnet/resnet_cifar.py
@@ -119,7 +119,7 @@ if __name__ == '__main__':
 
     epoch_size = args_opt.epoch_size
     net = resnet50(args_opt.num_classes)
-    ls = SoftmaxCrossEntropyWithLogits(sparse=True, is_grad=False, reduction="mean")
+    ls = SoftmaxCrossEntropyWithLogits(sparse=True, reduction="mean")
     opt = Momentum(filter(lambda x: x.requires_grad, net.get_parameters()), 0.01, 0.9)
 
     model = Model(net, loss_fn=ls, optimizer=opt, metrics={'acc'})