diff --git a/modules/callbacks.py b/modules/callbacks.py
index 4081140..2ae9d90 100644
--- a/modules/callbacks.py
+++ b/modules/callbacks.py
@@ -5,6 +5,8 @@ from threading import Thread
 import torch
 import transformers
 
+import modules.shared as shared
+
 
 # Copied from https://github.com/PygmalionAI/gradio-ui/
 class _SentinelTokenStoppingCriteria(transformers.StoppingCriteria):