Actual fix of the UI screen change