diff --git a/Configuration/config.ini b/Configuration/config.ini
index 6373a249e15389034550d0a0ac75bf8097d1c31e..64bba529b04897e70ebaee8328c7ad7275826921 100644
--- a/Configuration/config.ini
+++ b/Configuration/config.ini
@@ -6,7 +6,7 @@ symbols_vocab_size=26
 atom_vocab_size=17
 max_len_sentence=109
 max_atoms_in_sentence=1250
-max_atoms_in_one_type=324
+max_atoms_in_one_type=510
 
 [MODEL_ENCODER]
 dim_encoder = 768
diff --git a/Linker/__init__.py b/Linker/__init__.py
index c2a9483d03b868e0c2b00cae7a54f7bb7b7bd4db..b9380b473249eb38f1995474414ceb1eb6ea85ca 100644
--- a/Linker/__init__.py
+++ b/Linker/__init__.py
@@ -1,3 +1,4 @@
 from .Linker import Linker
 from .atom_map import atom_map
-from .AtomEmbedding import AtomEmbedding
\ No newline at end of file
+from .AtomEmbedding import AtomEmbedding
+from .AtomTokenizer import AtomTokenizer
\ No newline at end of file