Adjust timeout to match timeout in Java SDK