Forgot to remove the if