Fix for upgrader.