Set main correctly