diff --git a/dm/test/unittest/display_manager_test.cpp b/dm/test/unittest/display_manager_test.cpp
index b23b8009b262b87f36e86e38bd62d4390a69021c..ac00f864983a36f668b6765ef361c0fbaf7d09ba 100644
--- a/dm/test/unittest/display_manager_test.cpp
+++ b/dm/test/unittest/display_manager_test.cpp
@@ -2771,7 +2771,7 @@ HWTEST_F(DisplayManagerTest, GetPrimaryDisplaySystemDpi, TestSize.Level1)
 {
     float dpi = SingletonContainer::Get<DisplayManager>().GetPrimaryDisplaySystemDpi();
     SingletonContainer::Get<DisplayManagerAdapter>().InitDMSProxy();
-    sptr<DisplayInfo> displayInfo = new DisplayInfo();
+    sptr<DisplayInfo> displayInfo = SingletonContainer::Get<DisplayManagerAdapter>().GetPrimaryDisplayInfo();
     ASSERT_NE(displayInfo, nullptr);
     float primaryDisplaySystemDpi = displayInfo->GetDefaultVirtualPixelRatio();
     EXPECT_EQ(dpi, primaryDisplaySystemDpi);
diff --git a/dm/test/unittest/display_test.cpp b/dm/test/unittest/display_test.cpp
index 72b23341ad470dbfbad69d4a7286eef49e36327e..2db9bbee5388d424eff14113eab2a4713971d3d7 100644
--- a/dm/test/unittest/display_test.cpp
+++ b/dm/test/unittest/display_test.cpp
@@ -304,10 +304,12 @@ HWTEST_F(DisplayTest, GetLiveCreaseRegion, TestSize.Level1)
     } else {
         EXPECT_EQ(ret, DMError::DM_ERROR_DEVICE_NOT_SUPPORT);
     }
-    screenId = 5;
-    baseInfo->SetScreenId(screenId);
-    ret = disPlay->GetLiveCreaseRegion(region);
-    EXPECT_EQ(ret, DMError::DM_OK);
+    if (DisplayManager::GetInstance().IsFoldable()) {
+        screenId = 5;
+        baseInfo->SetScreenId(screenId);
+        ret = disPlay->GetLiveCreaseRegion(region);
+        EXPECT_EQ(ret, DMError::DM_OK);
+    }
 }
 
 /**