Review comment updates - remove extra newline, simplify assert in test