diff --git a/docs/optimizing-performance.md b/docs/optimizing-performance.md
index 91c5d515..a80193fe 100644
--- a/docs/optimizing-performance.md
+++ b/docs/optimizing-performance.md
@@ -79,11 +79,13 @@ class CounterButton extends React.Component {
   }
 
   render() {
-    <button
-      color={this.props.color}
-      onClick={() => this.setState(state => ({count: state.count + 1}))}>
-      Count: {this.state.count}
-    </button>
+    return (
+      <button
+        color={this.props.color}
+        onClick={() => this.setState(state => ({count: state.count + 1}))}>
+        Count: {this.state.count}
+      </button>
+    );
   }
 }
 ```
@@ -98,11 +100,13 @@ class CounterButton extends React.PureComponent {
   }
 
   render() {
-    <button
-      color={this.props.color}
-      onClick={() => this.setState(state => ({count: state.count + 1}))}>
-      Count: {this.state.count}
-    </button>
+    return (
+      <button
+        color={this.props.color}
+        onClick={() => this.setState(state => ({count: state.count + 1}))}>
+        Count: {this.state.count}
+      </button>
+    );
   }
 }
 ```