fixed to pass the tests