Fix up after NuGet API change