diff --git a/new_specie/print_detections.py b/new_specie/print_detections.py
index 0e1d86eb5e87481715a3b9bf243cd67a7f77e5a0..bd0b8b1c0e90fb0f866d9b3a2ad920a0d16dab1e 100755
--- a/new_specie/print_detections.py
+++ b/new_specie/print_detections.py
@@ -21,12 +21,13 @@ args = parser.parse_args()
 frontend = models.frontend_medfilt(args.SR, args.NFFT, args.sampleDur, args.nMel) if args.medfilt else models.frontend(args.SR, args.NFFT, args.sampleDur, args.nMel)
 df = pd.read_csv(args.detections)
 
+os.system('rm detections_pngs/*')
 loader = torch.utils.data.DataLoader(u.Dataset(df, args.audio_folder, args.SR, args.sampleDur + (2 if args.medfilt else 0)), batch_size=1, num_workers=8, collate_fn=u.collate_fn, shuffle=True)
 
 for x, idx in tqdm(loader):
     x = frontend(x).squeeze().detach()
     plt.imshow(x, origin='lower', aspect='auto', vmin=torch.quantile(x, .25), cmap='Greys', vmax=torch.quantile(x, .98))
     plt.subplots_adjust(top=1, bottom=0, left=0, right=1)
-    plt.savefig(f'annot_pngs/{idx.item()}')
+    plt.savefig(f'detections_pngs/{idx.item()}')
     plt.close()
 
diff --git a/new_specie/sort_cluster.py b/new_specie/sort_cluster.py
index a5d7d3351a893031af8c436fa44874dafb3e0ca9..b97b4ce624cfc64afa2cf1e0d07d4cc7870ea4d3 100755
--- a/new_specie/sort_cluster.py
+++ b/new_specie/sort_cluster.py
@@ -84,7 +84,7 @@ if args.umap_ndim == 2:
                 sig = signal.resample(sig, int(len(sig)/fs*args.SR))
             spec = frontend(torch.Tensor(sig).view(1, -1).float()).detach().squeeze()
             axSpec.clear()
-            axSpec.imshow(spec, origin='lower', aspect='auto')
+            axSpec.imshow(spec, origin='lower', aspect='auto', vmin=torch.quantile(spec, .25), cmap='Greys', vmax=torch.quantile(spec, .98))
             # Display and metadata management
             axSpec.set_title(f'{closest}, cluster {row.cluster:.0f} ({(df.cluster==row.cluster).sum()} points)')
             axScat.scatter(row.umap_x, row.umap_y, c='r')