could you add a unit test for this change?