diff --git a/subject_extract.py b/subject_extract.py
index c3ce293..6ecf54b 100644
--- a/subject_extract.py
+++ b/subject_extract.py
@@ -147,7 +147,7 @@ def __iter__(self):
 from keras.optimizers import Adam
 
 
-bert_model = load_trained_model_from_checkpoint(config_path, checkpoint_path)
+bert_model = load_trained_model_from_checkpoint(config_path, checkpoint_path, seq_len=None)
 
 for l in bert_model.layers:
     l.trainable = True