fix main method