Fix formatting from last refactoring