Oops bug fix