Fix main again