Can you please add a test for this method?