Update lib/colocated-babel-plugin.js

Co-authored-by: Robert Jackson <[email protected]>

Author: dfreeman

lib/colocated-babel-plugin.js

@@ -24,7 +24,7 @@ module.exports = function (babel) {
       ExportDefaultDeclaration(path, state) {
         let defaultExportDeclarationPath = path.get('declaration');
         let defaultExportIsExpressionOrClass =
-          defaultExportDeclarationPath.isClass() || defaultExportDeclarationPath.isExpression();
+          defaultExportDeclarationPath.isClass() || defaultExportDeclarationPath.isExpression() || defaultExportDeclarationPath.isFunction();
 
         if (
           state.colocatedTemplateFound !== true ||