Fix typo
Remove unnecessary class