Properly implement badge mocking