1 files changed, 7 insertions, 4 deletions
diff --git a/app/client/modules/samplernn/samplernn.datasets.js b/app/client/modules/samplernn/samplernn.datasets.js
index 58e6df4..bcc7878 100644
--- a/app/client/modules/samplernn/samplernn.datasets.js
+++ b/app/client/modules/samplernn/samplernn.datasets.js
@@ -66,15 +66,15 @@ class SampleRNNDatasets extends Component {
           linkFiles
           fileOptions={this.fileOptions}
           onPick={this.handlePick}
-          folder={samplernn.input.folder}
-          files={samplernn.input.files}
+          folder={samplernn.folder}
+          files={samplernn.folder.files}
         />
-        {this.renderData()}
       </div>
     )
   }
   renderData(){
-    const { samplernn } = this.props
+    // {this.renderData()}
+    const { samplernn, actions } = this.props
     if (samplernn.data === null) {
       return
     }
@@ -82,6 +82,9 @@ class SampleRNNDatasets extends Component {
       <div class='row params'>
         <FileList
           title='Folders'
+          className='folders'
+          fields='name'
+          onPick={actions.set_folder}
           files={samplernn.data.folders}
         />
         <FileList