Final bug fix