minor correction in main.cc