diff --git a/generate_function_doc.py b/generate_function_doc.py
index d5601ee2..1a5842f6 100755
--- a/generate_function_doc.py
+++ b/generate_function_doc.py
@@ -26,6 +26,7 @@
 import os
 import sys
 import yaml
+import re
 
 def main():
     """