Could you add a test for this change?