Fix tests to work with new shrinkwrap workflow