Fix test of main method