Fixing formatting after PR feedback.