gotta go with the dataset

1 files changed, 2 insertions, 2 deletions

diff --git a/app/client/dataset/dataset.loader.js b/app/client/dataset/dataset.loader.js
index 3887e69..ad42e46 100644
--- a/app/client/dataset/dataset.loader.js
+++ b/app/client/dataset/dataset.loader.js
@@ -83,7 +83,7 @@ export const load = module => {
       }
       const name = (file.name || 'unsorted').split('.')[0]
       const folder = folderLookup[file.folder_id] || unsortedFolder(module)
-      const dataset = getDataset(module, datasetLookup, name, folder, file.date || file.created_at)
+      const dataset = getDataset(module, datasetLookup, file.dataset || name, folder, file.date || file.created_at)
       if (file.url.match(file.name)) file.persisted = true
       dataset.input.push(file.id)
       folder.files.push(file.id)
@@ -95,7 +95,7 @@ export const load = module => {
       fileLookup[file.id] = file
       const pair = file.name.split('.')[0].split('-')
       const folder = folderLookup[file.folder_id] || unsortedFolder(module)
-      const dataset = getDataset(module, datasetLookup, pair[0], folder, file.date || file.created_at)
+      const dataset = getDataset(module, datasetLookup, file.dataset || pair[0], folder, file.date || file.created_at)
       dataset.output.push(file.id)
       folder.files.push(file.id)
       file.epoch = file.epoch || pair[1]