diff --git a/src/cli/gentest/templates/blockchain_test/transaction.py.j2 b/src/cli/gentest/templates/blockchain_test/transaction.py.j2
index ef192d37c07..6cef0e34dcd 100644
--- a/src/cli/gentest/templates/blockchain_test/transaction.py.j2
+++ b/src/cli/gentest/templates/blockchain_test/transaction.py.j2
@@ -8,7 +8,13 @@ from typing import Dict
 
 import pytest
 
-from ethereum_test_tools import Account, Environment, StateTestFiller, Storage, Transaction
+from ethereum_test_rpc.types import TransactionByHashResponse
+from ethereum_test_tools import (
+    Account,
+    Environment,
+    StateTestFiller,
+    Storage,
+)
 
 REFERENCE_SPEC_GIT_PATH = "N/A"
 REFERENCE_SPEC_VERSION = "N/A"