Solved error with UI references lost