From e926faf909f56a2e674e081d58b79a0dd1458460 Mon Sep 17 00:00:00 2001
From: streamich <streamich@gmail.com>
Date: Mon, 25 Feb 2019 12:13:23 +0100
Subject: [PATCH] =?UTF-8?q?fix:=20=F0=9F=90=9B=20new=20React=20behaviour?=
 =?UTF-8?q?=20needs=20useState=20to=20set=20something?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Issues: closes #128
---
 src/__stories__/useGetSet.story.tsx | 4 ++--
 src/useUpdate.ts                    | 5 ++++-
 2 files changed, 6 insertions(+), 3 deletions(-)

diff --git a/src/__stories__/useGetSet.story.tsx b/src/__stories__/useGetSet.story.tsx
index 2c2a7a9fe4..12dfae4947 100644
--- a/src/__stories__/useGetSet.story.tsx
+++ b/src/__stories__/useGetSet.story.tsx
@@ -32,9 +32,9 @@ const DemoWrong = () => {
 
 storiesOf('State/useGetSet', module)
   .add('Docs', () => <ShowDocs md={require('../../docs/useGetSet.md')} />)
-  .add('Demo', () =>
+  .add('Demo, 1s delay', () =>
     <Demo/>
   )
-  .add('DemoWrong', () =>
+  .add('DemoWrong, 1s delay', () =>
     <DemoWrong/>
   )
diff --git a/src/useUpdate.ts b/src/useUpdate.ts
index f6212617ba..0cf455148b 100644
--- a/src/useUpdate.ts
+++ b/src/useUpdate.ts
@@ -1,5 +1,8 @@
 import {useState} from 'react';
 
-const useUpdate = () => useState(0)[1] as (() => void);
+const useUpdate = () => {
+  const [, setState] = useState(0);
+  return () => setState(cnt => cnt + 1);
+};
 
 export default useUpdate;