diff --git a/IPython/core/ultratb.py b/IPython/core/ultratb.py
index 0caf7ec..9e7e8fb 100644
--- a/IPython/core/ultratb.py
+++ b/IPython/core/ultratb.py
@@ -101,7 +101,7 @@ import time
 import tokenize
 import traceback
 
-generate_tokens = tokenize.tokenize
+from tokenize import generate_tokens
 
 # For purposes of monkeypatching inspect to fix a bug in it.
 from inspect import getsourcefile, getfile, getmodule, \