Fixed #2250

Fixed #2250 with #2271 
No worries. I should have caught that earlier.