Fix for whitespace in code