Fix again test not passing