Forgot to update the if statements