Minor improvements to tests (remove old comments)