Fix treecall tests