fix if statement :/