updating so tests pass