fixed minor bug in main.c to make it compile