Removed 1 import line (for py2 only)