Updated to new NuGet package