added support for different upgrade variants