fix type bug

uncompleted type in main function