Forgot one more if statement check.