modify to pass test