Can you add a test with this change?