Fix typo
Fix tests