diff --git a/regression/xgboost_reg_main.cpp b/regression/xgboost_reg_main.cpp
index f8180ccc9..4e590b05d 100644
--- a/regression/xgboost_reg_main.cpp
+++ b/regression/xgboost_reg_main.cpp
@@ -111,7 +111,7 @@ namespace xgboost{
             inline void InitData( void ){
                 if( name_fmap != "NULL" ) fmap.LoadText( name_fmap.c_str() );
                 if( task == "dump" ) return;
-                if( task == "test" || task == "dumppath" ){
+                if( task == "pred" || task == "dumppath" ){
                     data.CacheLoad( test_path.c_str(), silent!=0, use_buffer!=0 );
                 }else{
                     // training