2 files changed, 7 insertions, 3 deletions

diff --git a/src/test/WixToolsetTest.Core.Native/CabinetFixture.cs b/src/test/WixToolsetTest.Core.Native/CabinetFixture.cs
index bc8f0ff8..fba0d5c9 100644
--- a/src/test/WixToolsetTest.Core.Native/CabinetFixture.cs
+++ b/src/test/WixToolsetTest.Core.Native/CabinetFixture.cs
@@ -2,11 +2,11 @@
 
 namespace WixToolsetTest.CoreNative
 {
-    using System;
     using System.IO;
     using System.Linq;
     using WixToolset.Core.Native;
     using WixToolsetTest.CoreNative.Utility;
+    using WixToolset.Data;
     using Xunit;
 
     public class CabinetFixture
@@ -22,7 +22,7 @@ namespace WixToolsetTest.CoreNative
                 var files = new[] { new CabinetCompressFile(TestData.Get(@"TestData\test.txt"), "test.txt") };
 
                 var cabinet = new Cabinet(cabPath);
-                cabinet.Compress(files, CabinetCompressionLevel.Low);
+                cabinet.Compress(files, CompressionLevel.Low);
 
                 Assert.True(File.Exists(cabPath));
             }
@@ -84,7 +84,7 @@ namespace WixToolsetTest.CoreNative
                     var files = new[] { new CabinetCompressFile(TestData.Get(@"TestData\test.txt"), "test.txt") };
 
                     var cabinet = new Cabinet(cabinetPath);
-                    cabinet.Compress(files, CabinetCompressionLevel.Low);
+                    cabinet.Compress(files, CompressionLevel.Low);
                 }
 
                 // Extract.
diff --git a/src/test/WixToolsetTest.Core.Native/WixToolsetTest.Core.Native.csproj b/src/test/WixToolsetTest.Core.Native/WixToolsetTest.Core.Native.csproj
index 47396ead..263712bb 100644
--- a/src/test/WixToolsetTest.Core.Native/WixToolsetTest.Core.Native.csproj
+++ b/src/test/WixToolsetTest.Core.Native/WixToolsetTest.Core.Native.csproj
@@ -17,6 +17,10 @@
   </ItemGroup>
 
   <ItemGroup>
+    <PackageReference Include="WixToolset.Data" Version="4.0.*" PrivateAssets="All" />
+  </ItemGroup>
+
+  <ItemGroup>
     <PackageReference Include="Microsoft.NET.Test.Sdk" Version="15.8.0" />
     <PackageReference Include="xunit" Version="2.4.1" />
     <PackageReference Include="xunit.runner.visualstudio" Version="2.4.1" PrivateAssets="All" />