aboutsummaryrefslogtreecommitdiffstatshomepage
path: root/tests/test_register.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/test_register.py')
-rw-r--r--tests/test_register.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/tests/test_register.py b/tests/test_register.py
index d053e9eb..0ce8351b 100644
--- a/tests/test_register.py
+++ b/tests/test_register.py
@@ -1,5 +1,5 @@
from infini.core import Core
-from infini.generator import Generator
+from infini.generator import TextGenerator
from infini.handler import Handler
from infini.input import Input
from infini.interceptor import Interceptor
@@ -38,7 +38,7 @@ def test_register():
pre_interceptor.interceptors = register.pre_interceptors
handler = Handler()
handler.handlers = register.handlers
- generator = Generator()
+ generator = TextGenerator()
generator.events = register.events
generator.global_variables = register.global_variables
interceptor = Interceptor()