fixed small typo
fixed unittest