Small readability refactor