Fixing unit tests for whitespace