I can't figure out how to use the action addon alongside this addon. It acts as if there is a forceReRender going on, because if I click fast enough, I can see my action text just for a split second. Here's my code:

import * as React from 'react';
import { storiesOf } from '@storybook/react';
import { action } from '@storybook/addon-actions';
import useState from 'storybook-addon-state';
import { Checkbox} from './Checkbox.tsx';

storiesOf('Forms', module)
  .addDecorator((storyFn) => (<div style={{ backgroundColor: '#fff', width: '450px', padding: '50px' }}>{storyFn()}</div>))
  .add('Checkbox', () => {
    const [checked, set] = useState('checked', true);
    const onChange = () => {
      action('onChange')('something');
      set(!checked)
    };

    return (
      <CustomCheckbox
        checked={checked}
        onChange={onChange} />
    );
  });

import addons from '@storybook/addons';
import { FORCE_RE_RENDER, STORY_CHANGED } from '@storybook/core-events';

let stores: Record<string, any> = {};

addons.getChannel().addListener(STORY_CHANGED, () => {
stores = {};
});

const capitalizeFirstLetter = (string: string) =>
string.charAt(0).toUpperCase() + string.slice(1);

export default function useState(
name: string,
store: T
): [T, (newState: T) => void] {
let actualVals = stores[name] === undefined ? store : stores[name];
stores[name] = actualVals;

const update = (newState: T) => {
actualVals= newState;
stores[name] = actualVals;
addons.getChannel().emit(FORCE_RE_RENDER);
};
update.toString = () => set${capitalizeFirstLetter(name)}(newState);

return [actualVals, update];
}