fix tests

The tests are currently broken, but only for very simple examples. This seems to happens when requiring the 1PI property. Failures can be found by comparing by hand the result with the expectation, as the auto-diff is not working properly.