Tiny fix to code formatting