final bug fix!