diff --git a/torchsig/utils/visualize.py b/torchsig/utils/visualize.py
index d1cd35b..36fb0fe 100755
--- a/torchsig/utils/visualize.py
+++ b/torchsig/utils/visualize.py
@@ -8,7 +8,6 @@
 import numpy as np
 import pywt
 import torch
-import pdbgit 
 
 
 class Visualizer:
@@ -923,7 +922,6 @@ def mask_class_to_outline(tensor: np.ndarray) -> Tuple[List[List[int]], List[Any
     for idx in range(batch_size):
         label = tensor[idx].numpy()
         class_idx_curr = []
-        pdb.set_trace()
         for individual_burst_idx in range(label.shape[0]):
             if np.count_nonzero(label[individual_burst_idx]) > 0:
                 class_idx_curr.append(individual_burst_idx)