diff --git a/auto_encoder_static.py b/auto_encoder_static.py
index 7c14d39..0b22efc 100644
--- a/auto_encoder_static.py
+++ b/auto_encoder_static.py
@@ -136,8 +136,8 @@ def train():
 
 
 def test():
-    load_model(model.encoder, 'checkpoints/' + config['name'] + '_enc', assert_consumed_on=True)
-    load_model(model.decoder, 'checkpoints/' + config['name'] + '_dec', assert_consumed_on=True)
+    load_model(model.encoder, 'checkpoints/' + config['name'] + '_enc')
+    load_model(model.decoder, 'checkpoints/' + config['name'] + '_dec')
     for batch in dataset_train:
         batch = static_dataset_train.expand_batch(batch_to_dict(batch, config['signature_dynamic']))