Add edge case test