BUG: fix collections in legend

closes #181