diff --git a/tests/test_plugin_run.py b/tests/test_plugin_run.py
index f5f910d5d5c9a9bb6b7591019857eb29fde35fe6..9c9d229229a2bd958b5c48d41a44a18520fdf7aa 100644
--- a/tests/test_plugin_run.py
+++ b/tests/test_plugin_run.py
@@ -18,14 +18,14 @@ import pytest
 
 from . import testutil
 
-from conservancy_beancount import plugin, _typing
+from conservancy_beancount import beancount_types, plugin
 
 CONFIG_MAP = {}
 HOOK_REGISTRY = plugin.HookRegistry()
 
 @HOOK_REGISTRY.add_hook
 class TransactionCounter:
-    DIRECTIVE = _typing.Transaction
+    DIRECTIVE = beancount_types.Transaction
     HOOK_GROUPS = frozenset()
 
     def run(self, txn):
@@ -34,7 +34,7 @@ class TransactionCounter:
 
 @HOOK_REGISTRY.add_hook
 class PostingCounter(TransactionCounter):
-    DIRECTIVE = _typing.Transaction
+    DIRECTIVE = beancount_types.Transaction
     HOOK_GROUPS = frozenset(['posting'])
 
     def run(self, txn):