diff --git a/src/account/PageStory.tsx b/src/account/PageStory.tsx
index 65bb37a..d4f8aad 100644
--- a/src/account/PageStory.tsx
+++ b/src/account/PageStory.tsx
@@ -1,6 +1,6 @@
 import type { DeepPartial } from "keycloakify/tools/DeepPartial";
 import type { KcContext } from "./KcContext";
-import { createGetKcContextMock } from "keycloakify/account";
+import { createGetKcContextMock } from "keycloakify/account/KcContext";
 import type {
     KcContextExtraProperties,
     KcContextExtraPropertiesPerPage
diff --git a/src/login/PageStory.tsx b/src/login/PageStory.tsx
index ca3388c..b167b77 100644
--- a/src/login/PageStory.tsx
+++ b/src/login/PageStory.tsx
@@ -1,7 +1,7 @@
 import type { DeepPartial } from "keycloakify/tools/DeepPartial";
 import type { KcContext } from "./KcContext";
 import KcApp from "./KcApp";
-import { createGetKcContextMock } from "keycloakify/login";
+import { createGetKcContextMock } from "keycloakify/login/KcContext";
 import type {
     KcContextExtraProperties,
     KcContextExtraPropertiesPerPage