Fixing the score estimator