Fix broken test

Oops, I'll fix that.