slight modification so that tests pass