diff --git a/src/author_identification.py b/src/author_identification.py
index 4369dc6..6c984cb 100755
--- a/src/author_identification.py
+++ b/src/author_identification.py
@@ -4,6 +4,7 @@ from data.dante_loader import load_latin_corpus, list_authors
 from data.features import *
 from model import AuthorshipVerificator
 from util.evaluation import f1_from_counters
+from sklearn.calibration import CalibratedClassifierCV
 import argparse
 
 AUTHORS_CORPUS_I = ['Dante', 'ClaraAssisiensis', 'GiovanniBoccaccio', 'GuidoFaba', 'PierDellaVigna']
@@ -63,8 +64,6 @@ def main():
             params = None
             C = args.C
 
-        from sklearn.calibration import CalibratedClassifierCV
-
         av = AuthorshipVerificator(C=C, params=params)
         av.fit(Xtr, ytr)