[TASK] Adapt to new wrapper method in Flux