Fix test after 55a54ae882a24ce2531361036eaa78404dd58992