chore: refactor

dhis2 · Dec 4, 2023 · 9f14e06 · 9f14e06
1 parent 5d8ebdf
commit 9f14e06
Showing 1 changed file with 10 additions and 6 deletions.
diff --git a/src/Map.js b/src/Map.js
@@ -486,19 +486,23 @@ export class MapGL extends Evented {
 
     // Add class to map container
     _addClass(className) {
-        const { classList } = this.getContainer()
+        if (this._mapgl) {
+            const { classList } = this._mapgl._container
 
-        if (!classList.contains(className)) {
-            classList.add(className)
+            if (!classList.contains(className)) {
+                classList.add(className)
+            }
         }
     }
 
     // Remove class from map container
     _removeClass(className) {
-        const { classList } = this.getContainer()
+        if (this._mapgl) {
+            const { classList } = this._mapgl._container
 
-        if (classList.contains(className)) {
-            classList.remove(className)
+            if (classList.contains(className)) {
+                classList.remove(className)
+            }
         }
     }
 }