Fix main.m