diff --git a/disk-mapper/internal/test/benchmark_test.go b/disk-mapper/internal/test/benchmark_test.go
index c1f3a37b3..28f995b4b 100644
--- a/disk-mapper/internal/test/benchmark_test.go
+++ b/disk-mapper/internal/test/benchmark_test.go
@@ -12,6 +12,8 @@ import (
 	"fmt"
 	"math"
 	"testing"
+  "log/slog"
+  "os"
 
 	"github.com/edgelesssys/constellation/v2/disk-mapper/internal/diskencryption"
 	"github.com/martinjungblut/go-cryptsetup"
diff --git a/disk-mapper/internal/test/integration_test.go b/disk-mapper/internal/test/integration_test.go
index 67ec65b8c..99f1e4d3b 100644
--- a/disk-mapper/internal/test/integration_test.go
+++ b/disk-mapper/internal/test/integration_test.go
@@ -18,6 +18,7 @@ import (
 	"strings"
 	"syscall"
 	"testing"
+  "log/slog"
 
 	"github.com/bazelbuild/rules_go/go/runfiles"
 	"github.com/edgelesssys/constellation/v2/disk-mapper/internal/diskencryption"