Added missing six import.