[UI] Support overriding data source classes