Fixed my bug fix