Fix issue with version updating