Fix Close

Can close the UI now,