fix bug ,