Work around lack of dict_keys index in data loader