diff --git a/saltax/contexts.py b/saltax/contexts.py
index 57fb625..c1c03b3 100644
--- a/saltax/contexts.py
+++ b/saltax/contexts.py
@@ -3,6 +3,7 @@
 import cutax
 import strax
 from immutabledict import immutabledict
+import pema
 
 # straxen XENONnT options/configuration
 XNT_COMMON_OPTS = straxen.contexts.xnt_common_opts.copy()
@@ -180,6 +181,9 @@ def xenonnt_salted(runid,
     st.set_config(dict(fax_file=instr_file_name))
     st.set_config(dict(saltax_mode=saltax_mode))
 
+    # Register pema plugins
+    st.register_all(pema.match_plugins)
+    
     return st
 
 def sxenonnt(runid=None,
diff --git a/saltax/match/visual.py b/saltax/match/visual.py
index 734d203..cef927c 100644
--- a/saltax/match/visual.py
+++ b/saltax/match/visual.py
@@ -188,4 +188,5 @@ def plot_wf(ind, st_salt, st_simu, st_data, runid, matched_simu,
                       matched_simu_s2_timerange_i[1]+s2_ext_window_samples)
     axs[1,1].set_xlabel("Time [10ns]")
     axs[1,1].legend()
-    fig.show()
\ No newline at end of file
+    fig.show()
+    
\ No newline at end of file
diff --git a/saltax/plugins/s_peaklets.py b/saltax/plugins/s_peaklets.py
index 2ab16de..456b4d9 100644
--- a/saltax/plugins/s_peaklets.py
+++ b/saltax/plugins/s_peaklets.py
@@ -876,4 +876,3 @@ def sum_waveform_salted(peaks, hits, records, record_links, adc_to_pe, n_top_cha
 
         p['n_saturated_channels'] = p['saturated_channel'].sum()
         p['area_per_channel'][:] = area_per_channel
-