fix in main method