make theme functionality works with customization plugins