Do you mind adding unit tests for this method?