Add catch for timeout