Add test for args validation