diff --git a/exercises/02.state-optimization/README.mdx b/exercises/02.state-optimization/README.mdx
index 9ede71c9..36ba1407 100644
--- a/exercises/02.state-optimization/README.mdx
+++ b/exercises/02.state-optimization/README.mdx
@@ -18,9 +18,9 @@ const [state, setState] = useState({ count: 0 })
 
 // ...
 setState({ count: 0 }) // <-- will trigger a rerender
-setState(previousState => {
+setState(previousState => ({
 	count: previousState.count
-}) // <-- will trigger a rerender
+})) // <-- will trigger a rerender
 setState(previousState => previousState) // <-- will not trigger a rerender
 ```