diff --git a/src/components/SonComponent.js b/src/components/SonComponent.js
index a62cafb..04b2d64 100644
--- a/src/components/SonComponent.js
+++ b/src/components/SonComponent.js
@@ -18,23 +18,22 @@ export default class SonComponent extends React.PureComponent {
         this.setState({
             ...this.state,
             showChildrenBorder: !this.state.showChildrenBorder
-        }, (state)=>{
+        }, ()=>{
             this.changeChildrenClass()
         })
     }
 
     changeChildrenClass = () => {
-        if(this.state.showChildrenBorder){
-            this.setState({
-                ...this.state,
-                childrenClass:[SonComponentCss.children]
-            })
-        } else {
-            this.setState({
-                ...this.state,
-                childrenClass:[]
-            })
+        let childrenClass = []
+        if (this.state.showChildrenBorder) {
+            childrenClass = [SonComponentCss.children]
+
         }
+
+        this.setState({
+            ...this.state,
+            childrenClass
+        })
     }
 
     render() {