Puzzle #61025 - "correct" solution is not best?

lichess.org/training/61025

In this puzzle there is a mate in 2 with Qxg4 Kh6 Qh4#. However it says this solution is "wrong" and insists on a mate in 3 starting with hxg4+. Is this a bug?

Thanks for the otherwise great training tool :)

On 1.Qxg4 Kh6 2. Qh4+ black has ...2 Qxh4 (no mate there)

woops, you're right.

i must have put it in the board editor wrong cause i could've sworn stockfish confirmed this for me. ignore this then :)

You don't need a board editor. There's an analysis board when you finish a puzzle.