Trivial refactor