Update main.cpp

"band-aid" as requested