@wangjian fix bug