Change the if statement for version