fix bug in plot_atn_loc