diff --git a/config/generators/component/stateless.js.hbs b/config/generators/component/stateless.js.hbs
index 2cb8d7c..cf6214f 100644
--- a/config/generators/component/stateless.js.hbs
+++ b/config/generators/component/stateless.js.hbs
@@ -26,8 +26,8 @@ function {{ properCase name }}(props: {}) {
     </Box>
   );
 }
-{{else if wantPropTypes}}
-{{#unless wantFlowTypes}}
+{{else}}
+{{#if wantPropTypes}}
 function {{ properCase name }}(props) {
   return (
     <Box>
@@ -35,7 +35,6 @@ function {{ properCase name }}(props) {
     </Box>
   );
 }
-{{/unless}}
 {{else}}
 function {{ properCase name }}() {
   return (
@@ -45,6 +44,7 @@ function {{ properCase name }}() {
   );
 }
 {{/if}}
+{{/if}}
 
 {{#if wantPropTypes}}
 {{ properCase name }}.propTypes = {