Nice change. The problem is, the unit test for it is testing the deprecated method now. Can you add a test for the new method?