It seems like comments are generally off by 1 line, e.g. <img width="1050" alt="image" src="https://user-images.githubusercontent.com/2694872/158846492-834c5b32-cd7a-49cf-abc9-979986290446.png"> This affects all test comments I have generated.