fix main method for test