Fixed an error in main.m