fix bug for main not passed