fix dataloader