fix main test