Fix tests comment location issue