fixing if statement...