fix main tests