Minor improvement of backprop