Fix update checking