[BUGFIX] Use class name to mark upgrade wizards done