diff --git a/src/nectarchain/tools/write_camera_calibration.py b/src/nectarchain/tools/write_camera_calibration.py
index e7fe4e7e..3c5234cd 100644
--- a/src/nectarchain/tools/write_camera_calibration.py
+++ b/src/nectarchain/tools/write_camera_calibration.py
@@ -42,11 +42,9 @@ class CalibrationHDF5Writer(Tool):
         800, help="Temporary cut on charge to eliminate events without led signal"
     ).tag(config=True)
 
-    pedestal_product = traits.enum_trait(
-        PedestalCalculator, default="PedestalIntegrator"
-    )
+    pedestal_product = traits.Enum(PedestalCalculator, default="PedestalIntegrator")
 
-    flatfield_product = traits.enum_trait(
+    flatfield_product = traits.Enum(
         FlatFieldCalculator, default="FlasherFlatFieldCalculator"
     )
 
diff --git a/src/nectarchain/tools/write_pedestals.py b/src/nectarchain/tools/write_pedestals.py
index 1d3dd257..5b6c7423 100644
--- a/src/nectarchain/tools/write_pedestals.py
+++ b/src/nectarchain/tools/write_pedestals.py
@@ -24,9 +24,7 @@ class PedestalHDF5Writer(Tool):
         config=True
     )
 
-    calculator_product = traits.enum_trait(
-        PedestalCalculator, default="PedestalIntegrator"
-    )
+    calculator_product = traits.Enum(PedestalCalculator, default="PedestalIntegrator")
 
     aliases = Dict(
         dict(