diff --git a/matrix_pygmalion_bot/ai/koboldcpp.py b/matrix_pygmalion_bot/ai/koboldcpp.py
index 7a497e6..964b8b8 100644
--- a/matrix_pygmalion_bot/ai/koboldcpp.py
+++ b/matrix_pygmalion_bot/ai/koboldcpp.py
@@ -12,8 +12,8 @@ import io
 import base64
 from PIL import Image, PngImagePlugin
 
-#from .pygmalion_helpers import get_full_prompt, num_tokens
-from .llama_helpers import get_full_prompt, num_tokens
+from .pygmalion_helpers import get_full_prompt, num_tokens
+#from .llama_helpers import get_full_prompt, num_tokens
 
 logger = logging.getLogger(__name__)