Fix an import bug in treecall