BUG: Added OrderedDict import to import statement