Fix: a minor bug for decorator.