Fix badge removal on collections page too

Grabbed from #722, seems to fix it.