Fix upgrade module.