Added a #ifdef TEST, and comment updated