fix  method to make test pass