diff --git a/wpilibcIntegrationTests/src/main/native/cpp/PCMTest.cpp b/wpilibcIntegrationTests/src/main/native/cpp/PCMTest.cpp
index 1cc70b7..ea271a5 100644
--- a/wpilibcIntegrationTests/src/main/native/cpp/PCMTest.cpp
+++ b/wpilibcIntegrationTests/src/main/native/cpp/PCMTest.cpp
@@ -35,7 +35,7 @@
   frc::DigitalInput m_fakeSolenoid2{TestBench::kFakeSolenoid2Channel};
 
   void Reset() {
-    m_pneumaticsModule.SetClosedLoopControl(false);
+    m_pneumaticsModule.DisableCompressor();
     m_fakePressureSwitch.Set(false);
   }
 };
@@ -46,7 +46,7 @@
 TEST_F(PCMTest, PressureSwitch) {
   Reset();
 
-  m_pneumaticsModule.SetClosedLoopControl(true);
+  m_pneumaticsModule.EnableCompressorDigital();
 
   // Turn on the compressor
   m_fakePressureSwitch.Set(true);