diff --git a/src/ligand_neighbourhood_alignment/update.py b/src/ligand_neighbourhood_alignment/update.py
index ba5f568..62803dc 100644
--- a/src/ligand_neighbourhood_alignment/update.py
+++ b/src/ligand_neighbourhood_alignment/update.py
@@ -231,6 +231,7 @@ def perform_all_alignments(
                     .assembly
                 ],
             )
+        if (datasets[dtag].mtz != "None") and (datasets[dtag].mtz is not None):
             mtz_path = datasets[dtag].mtz
             if mtz_path and mtz_path != "None":
                 xmap = read_xmap_from_mtz(mtz_path, "2Fo-Fc")
diff --git a/tests/test_integration.py b/tests/test_integration.py
index 23f01dd..30bedbf 100644
--- a/tests/test_integration.py
+++ b/tests/test_integration.py
@@ -39,7 +39,7 @@ def test_collator_upload_1(
 def test_aligner_upload_1(constants, upload_1_dir):
     log = str(Path(constants.TEST_DIR) / "aligner.log")
 
-    a = Aligner(constants.TEST_DIR, log_file=log, log_level=0)
+    a = Aligner(constants.TEST_DIR, )
     logger = a.logger
     utils.LOG = logger
 
@@ -97,7 +97,7 @@ def test_collator_upload_2(constants, config_2_file, upload_2_dir, uploads_dir,
 def test_aligner_upload_2(constants):
     log = str(Path(constants.TEST_DIR) / "aligner.log")
 
-    a = Aligner(constants.TEST_DIR, log_file=log, log_level=0)
+    a = Aligner(constants.TEST_DIR, )
     logger = a.logger
     utils.LOG = logger