fix a bug.

Signed-off-by: sunus <sunuslee@gmail.com>