Add test for the second argument