Made test pass again