Plex: Restore closure after setting boundary IDs in create_unit_square