Fix main.h