You don't take the rook. You checkmate. All four rooks are still on the board at the end of the puzzle. I think you were looking at the alternative solution for move 4, which, although the second best move under the circumstances, is inferior to checkmate by a large margin.