Update the comment in tests section