diff --git a/alphafold/model/modules.py b/alphafold/model/modules.py
index 266ed4e..7736897 100644
--- a/alphafold/model/modules.py
+++ b/alphafold/model/modules.py
@@ -556,10 +556,11 @@ class Attention(hk.Module):
 
     Arguments:
       q_data: A tensor of queries, shape [batch_size, N_queries, q_channels].
-      m_data: A tensor of memories from which the keys and values are
-        projected, shape [batch_size, N_keys, m_channels].
-      mask: A mask for the attention, shape [batch_size, N_queries, N_keys].
-      nonbatched_bias: Shared bias, shape [N_queries, N_keys].
+      m_data: A tensor of memories from which the keys and values are projected,
+        shape [batch_size, N_keys, m_channels].
+      mask: A mask for the attention, shape [batch_size, N_heads, N_queries,
+        N_keys].
+      nonbatched_bias: Shared bias, shape [N_heads, N_queries, N_keys].
 
     Returns:
       A float32 tensor of shape [batch_size, N_queries, output_dim].
diff --git a/run_alphafold.py b/run_alphafold.py
index 86a629d..c628bbe 100644
--- a/run_alphafold.py
+++ b/run_alphafold.py
@@ -89,8 +89,9 @@ flags.DEFINE_string('uniprot_database_path', None, 'Path to the Uniprot '
                     'database for use by JackHMMer.')
 flags.DEFINE_string('pdb70_database_path', None, 'Path to the PDB70 '
                     'database for use by HHsearch.')
-flags.DEFINE_string('pdb_seqres_database_path', None, 'Path to the PDB '
-                    'seqres database for use by hmmsearch.')
+flags.DEFINE_string('pdb_seqres_database_path', None, 'Full filepath to the '
+                    'PDB seqres database file (not just the directory) for use '
+                    'by hmmsearch.')
 flags.DEFINE_string('template_mmcif_dir', None, 'Path to a directory with '
                     'template mmCIF structures, each named <pdb_id>.cif')
 flags.DEFINE_string('max_template_date', None, 'Maximum template release date '