diff --git a/lib/server/middlewareButNotReally/storyPrivileges.ts b/lib/server/middlewareButNotReally/storyPrivileges.ts
index 7ba7d50..75dcfef 100644
--- a/lib/server/middlewareButNotReally/storyPrivileges.ts
+++ b/lib/server/middlewareButNotReally/storyPrivileges.ts
@@ -1,6 +1,7 @@
 import type { H3Event, EventHandlerRequest } from "h3";
 import { IStory } from "~/models/stories";
 import isLoggedIn from "~/lib/server/middlewareButNotReally/isLoggedIn";
+import { IDraft } from "~/models/stories/draft";
 export function canDelete(event: H3Event<EventHandlerRequest>, story: IStory) {
 	isLoggedIn(event);
 	return (
@@ -8,7 +9,10 @@ export function canDelete(event: H3Event<EventHandlerRequest>, story: IStory) {
 		story.author._id === event.context.currentUser?._id
 	);
 }
-export function canModify(event: H3Event<EventHandlerRequest>, story: IStory) {
+export function canModify(
+	event: H3Event<EventHandlerRequest>,
+	story: IStory | IDraft,
+) {
 	isLoggedIn(event);
 	return (
 		event.context.currentUser?._id === story.author._id ||