Fixing timeout in test.