Fix version check functionality