updated the test_args to (matrix, Lambda)
small variable change in tests