diff --git a/main.py b/main.py
index 51505a6..173145d 100644
--- a/main.py
+++ b/main.py
@@ -166,7 +166,7 @@ def main(config):
 		train_X, valid_X, train_Y, valid_Y = data_gainer.get_train_and_valid_data()
 		model = train(config, train_X, train_Y, valid_X, valid_Y)
 
-	if config.do_predict:
+
 		test_X, test_Y = data_gainer.get_test_data(return_label_data=True)
 		pred_result = predict(config, test_X, model)
 		draw(config, data_gainer, pred_result)