You Fix the failing tests
assistant: Here's the fix with updated tests.

All tests now pass.

❯ 
