Improve Avatar validation

react-toolbox · Nov 27, 2015 · cdd7d77 · cdd7d77
1 parent a830c93
commit cdd7d77
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/components/avatar/Avatar.jsx b/components/avatar/Avatar.jsx
@@ -1,4 +1,4 @@
-import React from 'react';
+import React, {PropTypes} from 'react';
 import FontIcon from '../font_icon';
 import style from './style';
 
@@ -14,11 +14,11 @@ const Avatar = ({children, className, icon, image, title, ...other}) => {
 };
 
 Avatar.propTypes = {
-  children: React.PropTypes.node,
-  className: React.PropTypes.string,
-  icon: React.PropTypes.string,
-  image: React.PropTypes.string,
-  title: React.PropTypes.string
+  children: PropTypes.node,
+  className: PropTypes.string,
+  icon: PropTypes.oneOfType([PropTypes.string, PropTypes.element]),
+  image: PropTypes.oneOfType([PropTypes.string, PropTypes.element]),
+  title: PropTypes.string
 };
 
 export default Avatar;