fix in main.cc