Add retry for timeout