Add tests for timeouts