Fix flattening of parameters