Changed back flatten(FX) -> vec(FX), #554