diff --git a/setup.py b/setup.py
index c5d02aad6..c40097702 100644
--- a/setup.py
+++ b/setup.py
@@ -7,6 +7,7 @@
 from setuptools.command.egg_info import egg_info
 from setuptools.command.bdist_egg import bdist_egg
 
+from io import BytesIO
 from subprocess import check_call
 
 import os