fix if statement tests