bug fix~