OS-2723 add missing callbacks in VM upgrade loop.