Typo
Fix dataset test