Fixed an error in the last bug fix