fix template for main
fix main