Fixup onUpgrade()