diff --git a/src/spikeanalysis/spike_plotter.py b/src/spikeanalysis/spike_plotter.py
index d3f7739..99e67f9 100644
--- a/src/spikeanalysis/spike_plotter.py
+++ b/src/spikeanalysis/spike_plotter.py
@@ -798,6 +798,8 @@ def plot_response_trace(
             current_length = stim_lengths[stimulus]
             current_bins = bins[stimulus]
 
+            response[~np.isfinite(response)]=np.nan
+
             if by_trial and by_neuron:
                 for neuron in range(np.shape(response)[0]):
                     for trial in range(np.shape(response)[1]):