Clean up CallExpression flattening logic.