Per expectations, it's a very trivial problem, but you have to pay super close details. They'll expect you to refactor like 1-line, among other nitpicky things, even if you make clean & concise, performant code.