Integrate Unit with DataLoader