diff --git a/weightslab/examples/PyTorch/ws-classification/main.py b/weightslab/examples/PyTorch/ws-classification/main.py
index 8f2ec0d0..1547e1f7 100644
--- a/weightslab/examples/PyTorch/ws-classification/main.py
+++ b/weightslab/examples/PyTorch/ws-classification/main.py
@@ -20,11 +20,9 @@
 
 from torchvision import datasets, transforms
 from torchmetrics.classification import Accuracy
-from torchvision import datasets, transforms
 from torch.utils.data import Dataset
 
 import weightslab as wl
-from weightslab.backend import ledgers
 from weightslab.baseline_models.pytorch.models import FashionCNN as CNN
 from weightslab.components.global_monitoring import (
     guard_training_context,