Fixed problem with dataLoader