[Fix] fix bug