Fix EE badge disappearing