fix bug.

Signed-off-by: Daogang Tang <daogangtang@gmail.com>