Fix upgrade too

Signed-off-by: emanuele <emanuele45@gmail.com>