Tiny refactoring to improve code clarity