main.c fix