Forgot to fix the weights as well.