Fix ui assigment