[Fix] bug fix