diff --git a/vpnor/test/force_readonly_toc.cpp b/vpnor/test/force_readonly_toc.cpp
index 026c976..e71b5d1 100644
--- a/vpnor/test/force_readonly_toc.cpp
+++ b/vpnor/test/force_readonly_toc.cpp
@@ -6,7 +6,7 @@
 #include "backend.h"
 #include "test/mbox.h"
 #include "test/system.h"
-#include "vpnor/pnor_partition_defs.h"
+#include "vpnor/ffs.h"
 }
 
 #include "vpnor/pnor_partition_table.hpp"
diff --git a/vpnor/test/toc_flags.cpp b/vpnor/test/toc_flags.cpp
index 13bd016..181c7b2 100644
--- a/vpnor/test/toc_flags.cpp
+++ b/vpnor/test/toc_flags.cpp
@@ -8,7 +8,7 @@
 #include <cassert>
 
 #include "common.h"
-#include "vpnor/pnor_partition_defs.h"
+#include "vpnor/ffs.h"
 
 static constexpr auto BLOCK_SIZE = 4 * 1024;
 static constexpr auto DATA_MASK = ((1 << 24) - 1);