Fix timing out test.