fixing dataset load issue