Add a test for this change