[Fix] Minor bug