Forgot to add the line I removed when testing out something with adam.