Forgot to change a line of code