Fixed some upgrade module bug