diff --git a/test/test_mocki2sslaveduplex.cpp b/test/test_mocki2sslaveduplex.cpp
index d8c8ea8..91cbc3c 100644
--- a/test/test_mocki2sslaveduplex.cpp
+++ b/test/test_mocki2sslaveduplex.cpp
@@ -9,11 +9,18 @@
 #include "i2sslaveduplex.hpp"
 #include "sdkwrapper.hpp"
 
+using ::testing::_;
 class MockI2sSlaveDuplexTest : public ::testing::Test {
  protected:
-  virtual void SetUp() { i2s = new rpp_driver::MockI2sSlaveDuplex(sdk_); }
+  virtual void SetUp() {
+    EXPECT_CALL(sdk_, pio_claim_unused_sm(_, _));
+    i2s = new rpp_driver::MockI2sSlaveDuplex(sdk_);
+  }
 
-  virtual void TearDown() { delete i2s; }
+  virtual void TearDown() {
+    EXPECT_CALL(sdk_, pio_sm_unclaim(_, _));
+    delete i2s;
+  }
 
   ::rpp_driver::MockSdkWrapper sdk_;
   ::rpp_driver::MockI2sSlaveDuplex *i2s;