diff --git a/IPython/core/ultratb.py b/IPython/core/ultratb.py
index e1f8057..0caf7ec 100644
--- a/IPython/core/ultratb.py
+++ b/IPython/core/ultratb.py
@@ -101,10 +101,7 @@ import time
 import tokenize
 import traceback
 
-try:  # Python 2
-    generate_tokens = tokenize.generate_tokens
-except AttributeError:  # Python 3
-    generate_tokens = tokenize.tokenize
+generate_tokens = tokenize.tokenize
 
 # For purposes of monkeypatching inspect to fix a bug in it.
 from inspect import getsourcefile, getfile, getmodule, \