Fix waiting for updates