Update tests to match new logfire API