if (type === 'FunctionExpression') {
          const isValidFunction = validateFunction(param.callee, param.params)
          const returnType = functions[param.callee].return

          if (typeof returnType === 'function') {