Fix loading on small dataset