Fixed test_arg_gen