diff --git a/test/gtest_imu.cpp b/test/gtest_imu.cpp
index 60d54ca54c8a6da097ae9b91381f47b2422b5b3d..6d51a5619ce1ff0615639be6d75ad3d7caf3089c 100644
--- a/test/gtest_imu.cpp
+++ b/test/gtest_imu.cpp
@@ -512,7 +512,7 @@ class Process_Factor_Imu : public testing::Test
             capture_imu = make_shared<CaptureImu>(t+dt, sensor_imu, data, sensor_imu->getNoiseCov());
             sensor_imu->process(capture_imu);
 
-            KF_1 = problem->getLastKeyFrame();
+            KF_1 = problem->getLastFrame();
             C_1  = KF_1->getCaptureList().front(); // front is Imu
             CM_1 = static_pointer_cast<CaptureMotion>(C_1);
 
diff --git a/test/gtest_processor_imu.cpp b/test/gtest_processor_imu.cpp
index 9ba48e91aee18e61ab938eba4179ad5de2ea7104..4099cf133f000dff71b7eb890c22a9c25c529605 100644
--- a/test/gtest_processor_imu.cpp
+++ b/test/gtest_processor_imu.cpp
@@ -175,7 +175,7 @@ TEST(ProcessorImu, voteForKeyFrame)
         - 1 KeyFrame at origin
         - 1 keyframe created by process() in voteForKeyFrame() since conditions to create a keyframe are met
     */
-    ASSERT_EQ(problem->getTrajectory()->getFrameList().size(),(unsigned int) 2);
+    ASSERT_EQ(problem->getTrajectory()->getFrameMap().size(),(unsigned int) 2);
 
     /* if max_time_span == 2,  Wolf tree should be