diff --git a/clmr_magnatagatune.py b/clmr_magnatagatune.py
index 2312c1d..7328bf9 100644
--- a/clmr_magnatagatune.py
+++ b/clmr_magnatagatune.py
@@ -18,7 +18,6 @@ import logging
 from pathlib import Path
 from typing import Union
 
-import subprocess
 import torchaudio
 import torch
 import numpy
@@ -31,6 +30,8 @@ sys.path.append(str(Path(__file__).parent))
 from utils.checkpoint import load_encoder_checkpoint
 from models.sample_cnn import SampleCNN
 
+log = logging.getLogger()
+
 
 @register(output_schema=['vec'])
 class ClmrMagnatagatune(NNOperator):