Allow restore to retry connecting to state server