final bug fix