Tweak code in CEThemeViewController