Dark theme fix