Fixup comment and improve test.