Fix wss module with partitioned domains

This fixes a bug which prevents the wss module from working in parallel. This happens because in parallel the index from the boundary expansions do not necessarily correspond to the boundary region id.