diff --git a/ocf_data_sampler/sample/base.py b/ocf_data_sampler/sample/base.py
index 491827ec..c5b44a37 100644
--- a/ocf_data_sampler/sample/base.py
+++ b/ocf_data_sampler/sample/base.py
@@ -74,7 +74,6 @@ def batch_to_tensor(batch: NumpyBatch) -> TensorBatch:
                 batch[k] = torch.as_tensor(v)
     return batch
 
-import torch
 
 def copy_batch_to_device(batch: dict, device: torch.device) -> dict:
     """