Fixed bug in decorator.