diff --git a/json-inspector.js b/json-inspector.js
index f9e9be6..909b65d 100644
--- a/json-inspector.js
+++ b/json-inspector.js
@@ -18,7 +18,10 @@ module.exports = React.createClass({
             React.PropTypes.array.isRequired,
         ]),
         // For now it expects a factory function, not element.
-        search: React.PropTypes.func,
+        search: React.PropTypes.oneOfType([
+            React.PropTypes.func.isRequired,
+            React.PropTypes.bool.isRequired,
+        ]),
         onClick: React.PropTypes.func,
         validateQuery: React.PropTypes.func,
         isExpanded: React.PropTypes.func,