Fix typo for small dataset