diff --git a/aidungeon/generator/gpt2/gpt2_generator.py b/aidungeon/generator/gpt2/gpt2_generator.py
index b153d88..660eebb 100644
--- a/aidungeon/generator/gpt2/gpt2_generator.py
+++ b/aidungeon/generator/gpt2/gpt2_generator.py
@@ -1,4 +1,4 @@
-from aidungeon.story.story.utils import *
+from aidungeon.story.utils import *
 import warnings
 warnings.filterwarnings("ignore")
 import os