fix int main() bug