Can you add some tests for this method?