Rehydrating Redux store when using systemjs hot reloader

app.js

import {createStore} from 'redux';
import reducer from './reducers/index.js'
import { rehydrate, rehydratingStore } from './utils/rehydratingStore.js';

const store = rehydratingStore()(createStore)(reducer);

export function __reload(deletedModule){
	const prevState = deletedModule.getState();
	debug('Reloaded. rehydrate with state: ', prevState.sketcher.objectsById);
	store.dispatch(rehydrate(prevState));
}
export function getState() {
	return store.getState();
}

rehydratingStore.js

const REHYDRATE = 'REHYDRATE';

export function rehydrate(state) {
  return { type: REHYDRATE, state };
};

export function rehydratingStore (config = {}) {
  return (next) => (reducer, initialState) => {
    const rehydrationReducer = createRehydrationReducer(reducer);

    const store = next(rehydrationReducer, initialState);
    const dispatch = store.dispatch;

    return {
      ...store,
      dispatch
    }
  }

  function createRehydrationReducer (reducer) {
    return (state, action) => {
      if (action.type === REHYDRATE) {
        let state = action.state;
        state = reducer(state, action)
        return state
      } else {
        return reducer(state, action);
      }
    }
  }
}