Fixed issue with UI registration