fixed compile error in main