diff --git a/docs/api/ShallowWrapper/dive.md b/docs/api/ShallowWrapper/dive.md
index 218fb51ff..b5f4aec7b 100644
--- a/docs/api/ShallowWrapper/dive.md
+++ b/docs/api/ShallowWrapper/dive.md
@@ -54,22 +54,28 @@ expect(wrapper.find(Bar).dive().find('.in-bar')).to.have.lengthOf(1);
- If you _only_ dive a `` (even if there is a `` in your component tree) you always get the default context value. If you want the `` to receive your ``'s `value` you must first `.dive()` the ``, the `.dive()` the ``.
```jsx
const { Provider, Consumer } = React.createContext('foo');
- class MyComponent extends React.Component {
- render() {
- return (
-
-
-
- {value => (
- {value}
- )}
-
-
-
- )
- }
+ function MyComponent() {
+ return (
+
+
+
+ {value => (
+ {value}
+ )}
+
+
+
+ );
}
const wrapper = shallow();
- wrapper.find(Consumer).dive().text(); // "foo"
- wrapper.find(Provider).dive().find(Consumer).dive().text(); // "bar"
+ wrapper
+ .find(Consumer)
+ .dive()
+ .text(); // "foo"
+ wrapper
+ .find(Provider)
+ .dive()
+ .find(Consumer)
+ .dive()
+ .text(); // "bar"
```