Revert previous commit because I just realized the code was correct.