BUG: remove six import