Fix if statement to be as should on CI.