Update tests, still failing
Inference tests pass