Small bugfix in main.c
Small bugfix in main.c