Fix prompt upgrade problem