Update comments in tests to be accurate