minor fix on test.cpp