1 files changed, 10 insertions, 12 deletions
diff --git a/app/client/modules/samplernn/samplernn.reducer.js b/app/client/modules/samplernn/samplernn.reducer.js
index 8d1ed74..5e83e98 100644
--- a/app/client/modules/samplernn/samplernn.reducer.js
+++ b/app/client/modules/samplernn/samplernn.reducer.js
@@ -1,20 +1,10 @@
 import types from '../../types'
 
 const samplernnInitialState = {
-  loading: false,
+  loading: true,
   error: null,
-  folder: {},
   folders: [],
-  datasets: [],
-  results: [],
-  input: {
-    folder: {},
-    files: [],
-  },
-  output: {
-    folder: {},
-    files: [],
-  },
+  folder: {},
   data: null,
 }
 
@@ -23,6 +13,7 @@ const samplernnReducer = (state = samplernnInitialState, action) => {
     case types.samplernn.init:
       return {
         ...state,
+        loading: false,
         data: action.data,
       }
     case types.socket.connect:
@@ -55,6 +46,13 @@ const samplernnReducer = (state = samplernnInitialState, action) => {
       return state
     case types.socket.status:
       return samplernnSocket(state, action.data)
+
+    case types.samplernn.set_folder:
+      return {
+        ...state,
+        folder: action.folder,
+      }
+
     default:
       return state
   }