Add tests for timeout