Add missed `import six`