Bug fix

Signed-off-by: Lingchao Xin <douglarek@gmail.com>