More VirtualRegisterStage work