diff --git a/pes_to_spec/model.py b/pes_to_spec/model.py
index be407b533520bfb55b93f626a2749fb7f1ff5e30..d86499c0905f70b50c5872a246555c04b6b53dd8 100644
--- a/pes_to_spec/model.py
+++ b/pes_to_spec/model.py
@@ -532,6 +532,10 @@ class Model(TransformerMixin, BaseEstimator):
         # size of the test subset
         self.validation_size = validation_size
 
+    def get_channels(self) -> List[str]:
+        """Get channels used in training."""
+        return self.x_model.named_steps["select"].channels
+
     def debug_peak_finding(self, low_res_data: Dict[str, np.ndarray], filename: str):
         """
         Produce image to understand if the peak finding step worked well.