Fixing comments in the test