Updating edge case tests