Use getArgument in the test