1 files changed, 3 insertions, 3 deletions
diff --git a/app/client/modules/samplernn/samplernn.results.js b/app/client/modules/samplernn/samplernn.results.js
index 85720cd..090ab21 100644
--- a/app/client/modules/samplernn/samplernn.results.js
+++ b/app/client/modules/samplernn/samplernn.results.js
@@ -46,19 +46,19 @@ class SampleRNNResults extends Component {
   }
   render(){
     if (this.props.samplernn.loading) return <Loading progress={this.props.samplernn.progress} />
-    const { folderLookup } = this.props.samplernn.data
+    const { folderLookup, fileLookup, datasetLookup } = this.props.samplernn.data
     // const { folderLookup } = samplernn
 
     const renders = Object.keys(folderLookup).sort(util.sort.stringSort.asc).map(key => {
       const folder = folderLookup[key]
       let { mapFn, sortFn } = util.sort.orderByFn('epoch desc')
-      const datasetPairs = folder.datasets.map(mapFn).sort(sortFn)
+      const datasetPairs = folder.datasets.map(name => datasetLookup[name]).map(mapFn).sort(sortFn)
       const bestRenders = datasetPairs
         .map(pair => pair[1])
         .filter(dataset => dataset.output.length)
         .map(dataset => {
           const { output } = dataset
-          return output.map(mapFn).sort(sortFn)[0][1]
+          return output.map(id => fileLookup[id]).map(mapFn).sort(sortFn)[0][1]
         })
       // console.log(bestRenders.map(r => r.epoch))
       const path = folder.name === 'unsorted'