Oops, bug fix