1 files changed, 3 insertions, 3 deletions
diff --git a/app/client/modules/samplernn/samplernn.show.js b/app/client/modules/samplernn/samplernn.show.js
index f16758a..8e271fc 100644
--- a/app/client/modules/samplernn/samplernn.show.js
+++ b/app/client/modules/samplernn/samplernn.show.js
@@ -20,13 +20,12 @@ class SampleRNNShow extends Component {
   render(){
     const { samplernn, match, history } = this.props
     const { folder } = samplernn || {}
-    console.log(folder)
     return (
       <div className='app'>
         <div class='heading'>
           <h1>{folder ? folder.name : 'Loading...'}</h1>
         </div>
-        {folder && folder.name !== 'unsorted' &&
+        {folder && folder.name && folder.name !== 'unsorted' &&
           <DatasetForm
             title='Add Files'
             module={samplernnModule}
@@ -35,7 +34,7 @@ class SampleRNNShow extends Component {
           />
         }
         <SampleRNNDatasets
-          id={this.props.match.params.id}
+          id={this.props.match.params.id || localStorage.getItem('samplernn.last_id')}
         />
       </div>
     )
@@ -44,6 +43,7 @@ class SampleRNNShow extends Component {
 
 const mapStateToProps = state => ({
   samplernn: state.module.samplernn,
+  dataset: state.dataset,
 })
 
 const mapDispatchToProps = (dispatch, ownProps) => ({