fix for badge not showing