#238 Fix upgrade command