Update for updated lib