Fix broken test introduced by #348