Minor bug fix

Properly fixed the whitespace bug.