Fix for update checking