Update to reflect package change