Fix failing tests from adding decorator.