diff --git a/plio/io/io_controlnetwork.py b/plio/io/io_controlnetwork.py index 0f96324fb6848ce31d069ed2985dcda2a6333cdb..57774f9670d728e9e6ad8224e4b86c332f06a496 100644 --- a/plio/io/io_controlnetwork.py +++ b/plio/io/io_controlnetwork.py @@ -264,8 +264,8 @@ class IsisStore(object): df_attr = self.point_field_map.get(attr, attr) if df_attr in g.columns: # As per protobuf docs for assigning to a repeated field. - if attr == 'aprioriCovar' or attr == 'adjustedCovar': - arr = g.iloc[0][attr] + if df_attr == 'aprioriCovar' or df_attr == 'adjustedCovar': + arr = g.iloc[0][df_attr] if isinstance(arr, np.ndarray): arr = arr.ravel().tolist()