Fix failing test

Oops.