Forgot an if statement removal