diff --git a/examples/react/CheckboxWithLabel.js b/examples/react/CheckboxWithLabel.js
index 9296cda1c5e0..0151f27745d8 100644
--- a/examples/react/CheckboxWithLabel.js
+++ b/examples/react/CheckboxWithLabel.js
@@ -14,8 +14,9 @@ export default class CheckboxWithLabel extends React.Component {
 
   render() {
     return (
-      <label>
+      <label ref={this.props.labelRef}>
         <input
+          ref={this.props.inputRef}
           type="checkbox"
           checked={this.state.isChecked}
           onChange={this.onChange}
diff --git a/examples/react/__tests__/CheckboxWithLabel-test.js b/examples/react/__tests__/CheckboxWithLabel-test.js
index e563307dab59..494504ca9cc5 100644
--- a/examples/react/__tests__/CheckboxWithLabel-test.js
+++ b/examples/react/__tests__/CheckboxWithLabel-test.js
@@ -1,24 +1,30 @@
 // Copyright 2004-present Facebook. All Rights Reserved.
 
-import React from 'react';
-import ReactDOM from 'react-dom';
+import React, {createRef} from 'react';
+
 import * as TestUtils from 'react-dom/test-utils';
 import CheckboxWithLabel from '../CheckboxWithLabel';
 
 it('CheckboxWithLabel changes the text after click', () => {
+  const checkboxLabelRef = createRef();
+  const checkboxInputRef = createRef();
   // Render a checkbox with label in the document
-  const checkbox = TestUtils.renderIntoDocument(
-    <CheckboxWithLabel labelOn="On" labelOff="Off" />
+  TestUtils.renderIntoDocument(
+    <CheckboxWithLabel
+      labelRef={checkboxLabelRef}
+      inputRef={checkboxInputRef}
+      labelOn="On"
+      labelOff="Off"
+    />
   );
 
-  const checkboxNode = ReactDOM.findDOMNode(checkbox);
+  const labelNode = checkboxLabelRef.current;
+  const inputNode = checkboxInputRef.current;
 
   // Verify that it's Off by default
-  expect(checkboxNode.textContent).toEqual('Off');
+  expect(labelNode.textContent).toEqual('Off');
 
   // Simulate a click and verify that it is now On
-  TestUtils.Simulate.change(
-    TestUtils.findRenderedDOMComponentWithTag(checkbox, 'input')
-  );
-  expect(checkboxNode.textContent).toEqual('On');
+  TestUtils.Simulate.change(inputNode);
+  expect(labelNode.textContent).toEqual('On');
 });