Can you make unit tests for that function?