def METHOD_NAME(
    exp: Expression,
) -> Expression:
    if (
        isinstance(exp, FunctionCall)
        and exp.function_name in AGGREGATION_FUNCTIONS
    ):
        return exp.transform(cast_column_to_nullable)
    return exp