In the methodology section it was assumed that centroids of atoms vectors after the encoder of VQGAE represent vectors of learned fragments. However, it is not possible to retrieve structure of a fragment by passing a single codebook vector to the decoder, as VQGAE will only generate a single atom. Therefore, we decided to extract fragments in an "indirect" manner. In this approach, substructures were extracted from molecules that contained atoms encoded in the same codebook vector, with these atoms serving as the centres of the substructures. The main assumption was that a small size of the environment around the central atom would result in the same extracted substructures that correspond to the learned fragment.