1 files changed, 6 insertions, 2 deletions
diff --git a/app/client/modules/samplernn/views/samplernn.import.js b/app/client/modules/samplernn/views/samplernn.import.js
index 6f5f78a..cce4aea 100644
--- a/app/client/modules/samplernn/views/samplernn.import.js
+++ b/app/client/modules/samplernn/views/samplernn.import.js
@@ -9,7 +9,7 @@ import Select from '../../../common/select.component'
 import TextInput from '../../../common/textInput.component'
 import Button from '../../../common/button.component'
 
-import SampleRNNDatasets from '../samplernn.datasets'
+import DatasetComponent from '../../../dataset/dataset.component'
 
 const samplernnModule = {
   name: 'samplernn',
@@ -90,7 +90,11 @@ class SampleRNNImport extends Component {
             </div>
           </div>
         </div>
-        <SampleRNNDatasets
+        <DatasetComponent
+          loading={samplernn.loading}
+          progress={samplernn.progress}
+          module={samplernnModule}
+          data={samplernn.data}
           id="unsorted"
           folder={folder}
           history={this.props.history}