diff --git a/enterprise/causal-clustering/src/test/java/org/neo4j/causalclustering/core/state/ClusterStateDirectoryTest.java b/enterprise/causal-clustering/src/test/java/org/neo4j/causalclustering/core/state/ClusterStateDirectoryTest.java
index d25008312149c..9f958d3c8b103 100644
--- a/enterprise/causal-clustering/src/test/java/org/neo4j/causalclustering/core/state/ClusterStateDirectoryTest.java
+++ b/enterprise/causal-clustering/src/test/java/org/neo4j/causalclustering/core/state/ClusterStateDirectoryTest.java
@@ -25,7 +25,7 @@
 
 import java.io.File;
 
-import org.neo4j.io.fs.DefaultFileSystemAbstraction;
+import org.neo4j.io.fs.FileSystemAbstraction;
 import org.neo4j.test.rule.TestDirectory;
 import org.neo4j.test.rule.fs.DefaultFileSystemRule;
 
@@ -36,7 +36,7 @@ public class ClusterStateDirectoryTest
 {
     @Rule
     public DefaultFileSystemRule fsRule = new DefaultFileSystemRule();
-    private DefaultFileSystemAbstraction fs = fsRule.get();
+    private FileSystemAbstraction fs = fsRule.get();
 
     @Rule
     public TestDirectory testDirectory = TestDirectory.testDirectory( fs );
diff --git a/enterprise/causal-clustering/src/test/java/org/neo4j/causalclustering/core/state/DumpClusterStateTest.java b/enterprise/causal-clustering/src/test/java/org/neo4j/causalclustering/core/state/DumpClusterStateTest.java
index ddbacdc2c4a5f..2e82c67a385b2 100644
--- a/enterprise/causal-clustering/src/test/java/org/neo4j/causalclustering/core/state/DumpClusterStateTest.java
+++ b/enterprise/causal-clustering/src/test/java/org/neo4j/causalclustering/core/state/DumpClusterStateTest.java
@@ -64,7 +64,7 @@ public class DumpClusterStateTest
     @Before
     public void setup() throws ClusterStateException
     {
-        clusterStateDirectory.initialize( fsa );
+        clusterStateDirectory.initialize( fsa.get() );
     }
 
     @Test