minor change in decorator