Another bug fix. Not sure how I missed this one