Fixing test that was relying on the order of dict.keys()