[main.cpp] minor change