Undo after game ended

When my opponent resigns, I like to finish the game vs a computer to see if I would really have been able to win the game. I like to try different moves and use the "undo" button a lot in this case. However, if I try something and blunder a stalemate, or the computer checkmates me, then the game just ends and I can't undo anymore. The only way I found out of this is to run a new analysis, go back to the second-to-last move, and then "finish vs computer" from there. That process is slow and cumbersome. It would be nice to undo moves even after the game has ended. That would also be very useful when doing drills.

There's a Finish vs Computer option on the game into tab.

Thanks for the reply Martin, but that doesn't solve the issue. First, I don't see a "game info" tab when the game ends vs computer. I only see that tab when the game ends vs human. But that's not the main issue I'm talking about.

I'm able to go to "Finish vs Computer" from a game that has ended (e.g. starting from a game from my Archive), and once I'm playing against the computer, I can try moves and undo them so I can try different lines. However, once the game ends against the computer (e.g. I blunder and the computer checkmates me in one move), I get the "Game Over" popup, and then I cannot undo moves anymore. It would be nice if we could undo moves and try different ones, even after the computer checkmates us.

Another example is when doing drills, e.g. the bishop+knight mate drill (https://www.chess.com/drills/practice/bishop-and-knight-mate). Sometimes I'll make mistakes and I can just undo them and play the correct moves instead. However, sometimes I blunder stalemate after 45 moves, and I get the "Draw by Stalemate" popup, and I cannot undo moves anymore and have to restart the drill from the beginning. I think it would be nice to undo mistakes even if they cause stalemate or checkmate.

I use the following:

--> Learn

--> Analysis

--> Set Up Position

--> choose your color as "White to move" or "Black to move" and configure the board in the way you would like to continue from

--> Load

--> Practice vs Computer (the dart board icon on the bottom right)

--> Change Bot (the robot head looking icon on the bottom right -- this is also optional)

Hope this helps!