Store proper dofmaps for parts (don't rely on set_current_part for global dofmap)