Fixed issue with UI initialized in init() method