fix main.cpp