Fixing a couple of comments