diff --git a/toolformer_pytorch/toolformer_pytorch.py b/toolformer_pytorch/toolformer_pytorch.py
index a58acb1..c2febab 100644
--- a/toolformer_pytorch/toolformer_pytorch.py
+++ b/toolformer_pytorch/toolformer_pytorch.py
@@ -1,4 +1,4 @@
-import re
+import regex as re
 
 from functools import partial, wraps
 from collections import namedtuple