small fix to edge case