Update for with-decorator

Signed-off-by: Julien Danjou <julien@danjou.info>