minor edge case fix