fix LCO specific tests

TOMToolkit · Sep 28, 2023 · 2072a6e · 2072a6e
1 parent 3fc2504
commit 2072a6e
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/tom_observations/facilities/lco.py b/tom_observations/facilities/lco.py
@@ -660,8 +660,9 @@ def __init__(self, *args, **kwargs):
                     label='Rotator Angle', required=False
                 )
                 # Add None option and help text for SOAR Gratings
-                self.fields[f'c_{j+1}_ic_{i+1}_grating'].help_text = 'Only for SOAR'
-                self.fields[f'c_{j+1}_ic_{i+1}_grating'].choices.insert(0, ('None', 'None'))
+                if self.fields.get(f'c_{j+1}_ic_{i+1}_grating', None):
+                    self.fields[f'c_{j+1}_ic_{i+1}_grating'].help_text = 'Only for SOAR'
+                    self.fields[f'c_{j+1}_ic_{i+1}_grating'].choices.insert(0, ('None', 'None'))
                 self.fields[f'c_{j+1}_ic_{i+1}_slit'].help_text = 'Only for Floyds'
 
     def convert_old_observation_payload_to_fields(self, data):