Small fix (adapt to defaultdict)