Fix the version class issue