Placing pytest_string in the right place and improving coverage.