Update main.cpp

- Bugfix