Fix test and corrected minor white space issue