Fix failing test

Missed this one.