Small fix to the theme plugin