Missed this bug

Fixed it best I could. I think it is fine