fixing the sample dataloader