Whitespace/code formatting fix