More edge case fixes