diff --git a/s3connector/s3_mock.go b/s3connector/s3_mock.go
index 5951079..34d7543 100644
--- a/s3connector/s3_mock.go
+++ b/s3connector/s3_mock.go
@@ -21,10 +21,10 @@ func NewS3Mock() S3Mock {
 	}
 }
 
-func (s3 *S3Mock) DeleteObject(key string) error {
+func (s3 *S3Mock) DeleteObject(objPath S3ObjectPath) error {
 	s3.mux.Lock()
 	defer s3.mux.Unlock()
-	delete(s3.storage, key)
+	delete(s3.storage, objPath.Key)
 	return nil
 }
 
diff --git a/s3connector/s3_mock_test.go b/s3connector/s3_mock_test.go
index 347c4d4..16a4cc7 100644
--- a/s3connector/s3_mock_test.go
+++ b/s3connector/s3_mock_test.go
@@ -71,8 +71,9 @@ func TestS3Mock_DeleteObject(t *testing.T) {
 	mock := NewS3Mock()
 	mock.storage["keyToDelete"] = "value"
 
+	objPath := S3ObjectPath{Key: "keyToDelete"}
 	// Test deleting existing object
-	err := mock.DeleteObject("keyToDelete")
+	err := mock.DeleteObject(objPath)
 	if err != nil {
 		t.Errorf("Failed to delete object: %s", err)
 	}