fix some issue in test