[TASK] Adapt to new wrapper method in Flux
Merge pull request #200 from bjo3rnf/compat

[TASK] Adapt to new wrapper method in Flux