Await registration with a timeout