Fix deprecated decorator usage