Refactor tests
Document edge case #1