Fix typo
Fix broken test