Oops
Fixed a bug I created when I fixed the previous bug.
Hope I don't create any bugs with this.