This made intuitive sense to me given that there is some overlap between manipulating objects in a 2D or 3D space and visualizing a math problem whether numeric or geometric.
I find it hard to believe that typical video game puzzles can help increase cognitive performance, but plausible that training spatial recognition might help train the same regions of the brain we use for math.
Even the basic mechanics: teaching math to my kids when they were young was purely a visual exercise of moving groups of things (coins, Cheerios, M&M's) around.