Fix in main.c