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