Bug fix

@lxyu