Fix test.cpp