You’re absolutely right and your solution is actually a better implementation.

I asked ChatGPT about this change and my version, with knowledge of how it and I work together, certainly prefers the old one.