diff --git a/tests/__tests__/mixins/labels.js b/tests/__tests__/mixins/labels.js
index 0a7a81f..3c3b92c 100644
--- a/tests/__tests__/mixins/labels.js
+++ b/tests/__tests__/mixins/labels.js
@@ -19,7 +19,7 @@ describe("Atom labels", () => {
         const basisAtomsNumber = wave.structure.basis.elements.length;
 
         atoms.forEach((atom) => {
-            const atomName = atom.name.split("-")[0];
+            const atomName = atom.nameWithLabel;
             const atomPosition = new THREE.Vector3().setFromMatrixPosition(atom.matrixWorld);
 
             const offsetVector = wave.getLabelOffsetVector(atomPosition, atomName);
@@ -49,7 +49,7 @@ describe("Atom labels", () => {
 
         const basisAtomsNumber = wave.structure.basis.elements.length;
         const doAllAtomsHaveLabels = atoms.every((atom) => {
-            const atomName = atom.name.split("-")[0];
+            const atomName = atom.nameWithLabel;
             const labelName = `labels-for-${atomName}`;
             const labelPointsByAtomName = labels.find(
                 (labelPoints) => labelPoints.name === labelName,