Fix documentation: unflatten instead of flatten