Mini refactor in formatter