Fix six import