Oops, fixed a silly bug