Update close behavior