diff --git a/applications/apertif/apertif_unb1_correlator/tb/python/tc_wpfb_src_out_arr.py b/applications/apertif/apertif_unb1_correlator/tb/python/tc_wpfb_src_out_arr.py
index 874c372e0938a2a511424beb576ccaa036480189..06a6a49567e24bad29d6d33df9e10790c5f3f6d8 100644
--- a/applications/apertif/apertif_unb1_correlator/tb/python/tc_wpfb_src_out_arr.py
+++ b/applications/apertif/apertif_unb1_correlator/tb/python/tc_wpfb_src_out_arr.py
@@ -1,5 +1,6 @@
 
 from common import *
+from common_dsp import *
 from file_io import *
 import os
 import matplotlib
@@ -23,7 +24,7 @@ complex_list = rec_file_to_complex(REC_FILE, complex_width=WPFB_OUT_DATA_WIDTH)
 f = deinterleave(complex_list[5], 2, 1)[0]
 
 T = 1.0 / NOF_CHANNELS
-yf = f
+yf = fft_reorder(f)
 xf = range(NOF_CHANNELS)
 xf = fftshift(xf)
 yplot = fftshift(yf)