Can you add a test case for this new method?