Updated unit tests to be compatible with the new recover function