diff --git a/drm/1.0/vts/functional/drm_hal_vendor_test.cpp b/drm/1.0/vts/functional/drm_hal_vendor_test.cpp
index 90608b9..0ea35c4 100644
--- a/drm/1.0/vts/functional/drm_hal_vendor_test.cpp
+++ b/drm/1.0/vts/functional/drm_hal_vendor_test.cpp
@@ -591,14 +591,17 @@
  */
 TEST_P(DrmHalVendorPluginTest, RestoreKeysClosedSession) {
     for (auto config : contentConfigurations) {
-        auto sessionId = openSession();
-        hidl_vec<uint8_t> keySetId = loadKeys(sessionId, config);
-        EXPECT_NE(0u, keySetId.size());
-        closeSession(sessionId);
-        sessionId = openSession();
-        closeSession(sessionId);
-        Status status = drmPlugin->restoreKeys(sessionId, keySetId);
-        EXPECT_EQ(Status::ERROR_DRM_SESSION_NOT_OPENED, status);
+        if (config.policy.allowOffline) {
+            auto sessionId = openSession();
+            hidl_vec<uint8_t> keySetId =
+                    loadKeys(sessionId, config, KeyType::OFFLINE);
+            EXPECT_NE(0u, keySetId.size());
+            closeSession(sessionId);
+            sessionId = openSession();
+            closeSession(sessionId);
+            Status status = drmPlugin->restoreKeys(sessionId, keySetId);
+            EXPECT_EQ(Status::ERROR_DRM_SESSION_NOT_OPENED, status);
+        }
     }
 }