Fix js flatten/unflatten