diff --git a/images/agent/src/internal/utils/volume_cleanup_ce.go b/images/agent/src/internal/utils/volume_cleanup_ce.go
index 7992625b..2d8e163d 100644
--- a/images/agent/src/internal/utils/volume_cleanup_ce.go
+++ b/images/agent/src/internal/utils/volume_cleanup_ce.go
@@ -22,6 +22,6 @@ import (
 	"agent/internal/logger"
 )
 
-func VolumeCleanup(_ context.Context, _ logger.Logger, _ BlockDeviceOpener, _, _, _ string) error {
+func VolumeCleanup(_ context.Context, _ logger.Logger, _ BlockDeviceOpener, _, _, _ string, _ *RangeCover) error {
 	return fmt.Errorf("volume cleanup is not supported in your edition")
 }
diff --git a/images/agent/src/internal/utils/volume_cleanup_ce_test.go b/images/agent/src/internal/utils/volume_cleanup_ce_test.go
index c9b02353..268e03a0 100644
--- a/images/agent/src/internal/utils/volume_cleanup_ce_test.go
+++ b/images/agent/src/internal/utils/volume_cleanup_ce_test.go
@@ -33,7 +33,7 @@ var _ = Describe("Cleaning up volume", func() {
 		Expect(err).NotTo(HaveOccurred())
 	})
 	It("Unsupported", func() {
-		err := VolumeCleanup(context.Background(), log, nil, "", "", "")
+		err := VolumeCleanup(context.Background(), log, nil, "", "", "", nil)
 		Expect(err).To(MatchError("volume cleanup is not supported in your edition"))
 	})
 })