Minor improvement to backprop