Fix check for version