Fix that colorpicker shows up on click

storybookjs · Sep 24, 2018 · ccfb1f1 · ccfb1f1
1 parent dd1953c
commit ccfb1f1
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/addons/knobs/src/components/types/Color.js b/addons/knobs/src/components/types/Color.js
@@ -28,10 +28,13 @@ class ColorType extends React.Component {
     document.addEventListener('mousedown', this.handleWindowMouseDown);
   }
 
-  shouldComponentUpdate(nextProps) {
+  shouldComponentUpdate(nextProps, nextState) {
     const { knob } = this.props;
+    const { displayColorPicker } = this.state;
 
-    return nextProps.knob.value !== knob.value;
+    return (
+      nextProps.knob.value !== knob.value || nextState.displayColorPicker !== displayColorPicker
+    );
   }
 
   componentWillUnmount() {