Fix compilation of main.cpp