Tiny refactoring

Yep