fix code

Signed-off-by: coldnew <coldnew.tw@gmail.com>