diff --git a/website/create_code_samples b/website/create_code_samples
index 659a3d71c..2b9938edb 100755
--- a/website/create_code_samples
+++ b/website/create_code_samples
@@ -4,6 +4,7 @@ from __future__ import unicode_literals
 import os
 import ast
 import io
+import re
 
 import plac