Fix last commit.

Signed-off-by: Kevin <kevainxxx@gmail.com>