Make sure update box is updated after click