WriteSet now returns a proper map