Add a test to verify the fix works