This should make test pass again