diff --git a/bin/baseline_sctm_scores_setup.py b/bin/baseline_sctm_scores_setup.py
index ee727ff..d862317 100644
--- a/bin/baseline_sctm_scores_setup.py
+++ b/bin/baseline_sctm_scores_setup.py
@@ -13,7 +13,7 @@ following commands:
 > conda activate omegafold
 > python bin/omegafold_across_gpus.py sctm_baseline_esm_residues/*.fasta -o sctm_baseline_omegafold_predictions
 > conda activate protdiff
-> python bin/omegafold_self_tm.py -p sctm_baseline_esm_residues -f sctm_baseline_omegafold_predictions -o baseline_sctm_scores.json -p baseline_sctm_scores.pdf
+> python bin/omegafold_self_tm.py -p sctm_baseline_real_pdbs -f sctm_baseline_omegafold_predictions -o baseline_sctm_scores
 """
 import logging
 import os
diff --git a/bin/omegafold_self_tm.py b/bin/omegafold_self_tm.py
index 5d8de04..89837de 100644
--- a/bin/omegafold_self_tm.py
+++ b/bin/omegafold_self_tm.py
@@ -77,7 +77,7 @@ def main():
     args = parser.parse_args()
 
     assert os.path.isdir(args.folded)
-    assert os.path.isdir(args.predicted)
+    assert os.path.isdir(args.predicted), f"Directory not found: {args.predicted}"
 
     orig_predicted_backbones = glob(os.path.join(args.predicted, "*.pdb"))
     logging.info(