Missed another line of code, had to add