Fix whitespace bug when displaying code