Fix minor bug.

Signed-off-by: tungpham31 <tungpham31@gmail.com>