Fixed a just-broken test