Fix closing ui part