Update test_kb.cpp