Following https://github.com/tskit-dev/tsinfer/pull/959/commits/57f9209dcca48fbe70d397dba67511f4def3ae49 #959