Update tests for new _stats API