diff --git a/testsr.py b/testsr.py
index f48dd6acc..558a944c5 100755
--- a/testsr.py
+++ b/testsr.py
@@ -4,8 +4,7 @@ import json
 import re
 import sys
 
-from aider.coders.editblock_coder import DEFAULT_FENCE, find_original_update_blocks
-
+from aider.coders.editblock_coder import DEFAULT_FENCE, find_original_update_blocks, all_fences
 
 def process_markdown(filename):
     try: