if statement fixed