Fix badges getting

can't really unit test that until editentity
doen't support badges. Tested locally though
and works.

Change-Id: Icec1ad6ac90523b7f7fc41e9d003d75b9f04736d