Small theme fix