Magic Sort is the kind of puzzle game that feels harmless because each level is short. That is exactly why it needs a stopping rule.
Choose a level count before opening the game: one puzzle, three puzzles, or ten minutes. Add Magic Sort to Screen Time and use Sandflow so the active session stays visible.
When the timer reaches your planned limit, close the game before starting the next puzzle.