diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml
index 684326064..d8ca5400a 100644
--- a/.github/workflows/tests.yml
+++ b/.github/workflows/tests.yml
@@ -34,7 +34,7 @@ jobs:
       fail-fast: false
       matrix:
         python_version: ["3.10", "3.11"]
-    timeout-minutes: 10
+    timeout-minutes: 20
 
     steps:
       - name: Check out repository code
diff --git a/src/axolotl/prompt_strategies/dpo/__init__.py b/src/axolotl/prompt_strategies/dpo/__init__.py
index 1a149f452..7f5e6eb64 100644
--- a/src/axolotl/prompt_strategies/dpo/__init__.py
+++ b/src/axolotl/prompt_strategies/dpo/__init__.py
@@ -5,4 +5,4 @@ from functools import partial
 
 from ..base import load as load_base
 
-load = partial(load_base, module="axolotl.prompt_strategies.dpo")
+load = partial(load_base, module_base="axolotl.prompt_strategies.dpo")
diff --git a/src/axolotl/prompt_strategies/orpo/chat_template.py b/src/axolotl/prompt_strategies/orpo/chat_template.py
index fb39bcf8f..9953fe87e 100644
--- a/src/axolotl/prompt_strategies/orpo/chat_template.py
+++ b/src/axolotl/prompt_strategies/orpo/chat_template.py
@@ -36,6 +36,7 @@ def load(
             chat_template = chat_templates(chat_template)
         except ValueError:
             pass
+    tokenizer.chat_template = chat_template
 
     return ORPOTokenizingStrategy(
         ORPOPrompter(chat_template, tokenizer),