fix some of compilation error in tests