diff --git a/sql/core/src/test/scala/org/apache/spark/sql/execution/streaming/state/StateStoreSuite.scala b/sql/core/src/test/scala/org/apache/spark/sql/execution/streaming/state/StateStoreSuite.scala
index 492ea1a06f9c..f16aa013c43c 100644
--- a/sql/core/src/test/scala/org/apache/spark/sql/execution/streaming/state/StateStoreSuite.scala
+++ b/sql/core/src/test/scala/org/apache/spark/sql/execution/streaming/state/StateStoreSuite.scala
@@ -2736,19 +2736,17 @@ abstract class StateStoreSuiteBase[ProviderClass <: StateStoreProvider]
   }
 
   test("SPARK-54063: forceSnapshot metric populated when shouldForceSnapshotOnCommit is true") {
-    withTempDir { dir =>
-      tryWithProviderResource(newStoreProvider()) { provider =>
-        val store = provider.getStore(0, forceSnapshotOnCommit = true)
-        put(store, "a", 0, 1)
-        store.commit()
-        // Verify that a snapshot file was created for version 1
-        val metricPair = store.metrics.customMetrics.find { case (metric, _) =>
-          metric.name.contains("rocksdbForceSnapshotCount") ||
-          metric.name.contains("forceSnapshotCount")
-        }
-        assert(metricPair.isDefined)
-        assert(metricPair.get._2 == 1L, s"forceSnapshot should be 1 but was ${metricPair.get._2}")
+    tryWithProviderResource(newStoreProvider()) { provider =>
+      val store = provider.getStore(0, forceSnapshotOnCommit = true)
+      put(store, "a", 0, 1)
+      store.commit()
+      // Verify that a snapshot file was created for version 1
+      val metricPair = store.metrics.customMetrics.find { case (metric, _) =>
+        metric.name.contains("rocksdbForceSnapshotCount") ||
+        metric.name.contains("forceSnapshotCount")
       }
+      assert(metricPair.isDefined)
+      assert(metricPair.get._2 == 1L, s"forceSnapshot should be 1 but was ${metricPair.get._2}")
     }
   }