Update for new package