Fix program running issue