diff --git a/src/Icon.tsx b/src/Icon.tsx
index 7f153e0..b79352c 100644
--- a/src/Icon.tsx
+++ b/src/Icon.tsx
@@ -18,12 +18,10 @@ export interface IconProps {
   type: ActionType;
 }
 
-export default class Icon extends React.Component<IconProps, any> {
-  render() {
-    let prefixCls = 'react-viewer-icon';
+export default function Icon(props: IconProps) {
+  let prefixCls = 'react-viewer-icon';
 
-    return (
-      <i className={`${prefixCls} ${prefixCls}-${ActionType[this.props.type]}`}></i>
-    );
-  }
+  return (
+    <i className={`${prefixCls} ${prefixCls}-${ActionType[props.type]}`}></i>
+  );
 }