fix the main.m