diff --git a/data/model/dcrnn_test_config.yaml b/data/model/dcrnn_test_config.yaml
index 93cbe59..182ff5c 100644
--- a/data/model/dcrnn_test_config.yaml
+++ b/data/model/dcrnn_test_config.yaml
@@ -25,7 +25,7 @@ model:
 train:
   base_lr: 0.01
   dropout: 0
-  epoch: 44
+  epoch: 51
   epochs: 100
   epsilon: 1.0e-3
   global_step: 0
diff --git a/run_demo.py b/run_demo.py
index 05b617f..3f07617 100644
--- a/run_demo.py
+++ b/run_demo.py
@@ -32,6 +32,6 @@ if __name__ == '__main__':
     parser.add_argument('--use_cpu_only', default=False, type=str, help='Whether to run tensorflow on cpu.')
     parser.add_argument('--config_filename', default='data/model/pretrained/METR-LA/config.yaml', type=str,
                         help='Config file for pretrained model.')
-    parser.add_argument('--output_filename', default='data/dcrnn_predictions.npz')
+    parser.add_argument('--output_filename', default='data/dcrnn_predictions_tf.npz')
     args = parser.parse_args()
     run_dcrnn(args)