seems like backprop is fixed but grad check is not