fix missing include in main.cpp