Forgot to add a line of code