diff --git a/bin/your_heimdall.py b/bin/your_heimdall.py
index 584da21..9751778 100644
--- a/bin/your_heimdall.py
+++ b/bin/your_heimdall.py
@@ -173,7 +173,7 @@
             )
 
     logging.info("Input Arguments:-")
-    for arg, value in sorted(vars(args).items()):
+    for arg, value in sorted(vars(values).items()):
         logging.info("%s: %r", arg, value)
 
     your_object = Your(file=values.files)
diff --git a/bin/your_rfimask.py b/bin/your_rfimask.py
index 4a091c3..7ac801a 100644
--- a/bin/your_rfimask.py
+++ b/bin/your_rfimask.py
@@ -2,6 +2,7 @@
 
 import argparse
 import logging
+from datetime import datetime
 
 import numpy as np
 from rich.logging import RichHandler