Fix error with update checking