Forgot to add an updated if statement