fixbug

Signed-off-by: Neo Chan <neo.chan@live.com>