Fix in main.cpp