Small fix in test.cpp