Fix test failure

Regression from d5c8d0f