Forgot this in the refactor.