Focus maintained while resetting UI