Version upgrade fix