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

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