diff --git a/profold2/common/residue_constants.py b/profold2/common/residue_constants.py
index 1c249ff4..a35b00d7 100644
--- a/profold2/common/residue_constants.py
+++ b/profold2/common/residue_constants.py
@@ -859,6 +859,10 @@ def make_bond_key(atom1_name, atom2_name):
 atom_type_num = len(atom_types)  # := 37.
 atom14_type_num = env('profold2_atom14_type_num', defval=14, dtype=int)  # := 14.
 
+prot_atom_type_num = 14
+dna_atom_type_num  = 22
+rna_atom_type_num  = 23
+
 # A compact atom encoding with 14 columns
 # pylint: disable=line-too-long
 # pylint: disable=bad-whitespace