Update and correct so tests pass