Fixed a residual bug from when I refactored things