Bug fix

Signed-off-by: Ken <kencplee@gmail.com>