fix return of main