diff --git a/packages/data-entry/src/test-utils/mocks.ts b/packages/data-entry/src/test-utils/mocks.ts
index d3a4eb8..d1208e0 100644
--- a/packages/data-entry/src/test-utils/mocks.ts
+++ b/packages/data-entry/src/test-utils/mocks.ts
@@ -1,11 +1,10 @@
 import { Theme } from '@mui/material';
 import { MainProps as ApplicationContext } from '../components';
 import { MockVaultParams, Vault, createVaultMock } from 'obsidian';
-import { Directory } from '@waynevanson/obsidian-mocks/dist/vault/data-adapter';
 
 export interface MockOptions {
   application?: Partial<Omit<ApplicationContext, 'vault'>> & {
-    vault: Directory;
+    vault: MockVaultParams;
   };
 }
 
diff --git a/packages/data-entry/src/test-utils/rtl.tsx b/packages/data-entry/src/test-utils/rtl.tsx
index 2fcdc22..42a8448 100644
--- a/packages/data-entry/src/test-utils/rtl.tsx
+++ b/packages/data-entry/src/test-utils/rtl.tsx
@@ -47,7 +47,7 @@ export function render<
     {} as never,
     options?.application ?? {},
   );
-  const mocks = createMocks({ vault: { root: vaultOptions } });
+  const mocks = createMocks({ vault: vaultOptions });
   const application = { ...most, vault: mocks.vault };
 
   const wrapper: React.JSXElementConstructor<{