Final bug fix!