Small fix in decorator.