Fix upgrade bug - thx for report Alexander Tzokev