fix large dataset issue