diff --git a/pyshexc/parser/ShExDocLexer.py b/pyshexc/parser/ShExDocLexer.py
index 1edef1d..dbd6fc4 100644
--- a/pyshexc/parser/ShExDocLexer.py
+++ b/pyshexc/parser/ShExDocLexer.py
@@ -1,9 +1,11 @@
 # Generated from ShExDoc.g4 by ANTLR 4.9
 from antlr4 import *
 from io import StringIO
-from typing import TextIO
 import sys
-
+if sys.version_info[1] > 5:
+    from typing import TextIO
+else:
+    from typing.io import TextIO
 
 
 def serializedATN():