Add missing import of OrderedDict