diff --git a/skywipe/safeguard.py b/skywipe/safeguard.py
index c3b7924..d25d1dc 100644
--- a/skywipe/safeguard.py
+++ b/skywipe/safeguard.py
@@ -1,17 +1,21 @@
 """Safeguard module for Skywipe"""
 
 import sys
+import logging
+from typing import Optional
 from .logger import get_logger
 
 
 CONFIRM_RESPONSES = {"yes", "y"}
 
 
-def require_confirmation(operation: str, skip_confirmation: bool = False) -> None:
+def require_confirmation(operation: str, skip_confirmation: bool = False, logger: Optional[logging.Logger] = None) -> None:
     if skip_confirmation:
         return
 
-    logger = get_logger()
+    if logger is None:
+        logger = get_logger()
+
     logger.warning(f"This will {operation}")
     logger.warning("This operation is DESTRUCTIVE and cannot be undone!")