Fix wonky test