diff --git a/Logic/FilteredLayer.ts b/Logic/FilteredLayer.ts
index 87ba7c6..6ee431b 100644
--- a/Logic/FilteredLayer.ts
+++ b/Logic/FilteredLayer.ts
@@ -110,16 +110,6 @@ export class FilteredLayer {
     }
 
 
-    public updateStyle() {
-        if (this._geolayer === undefined) {
-            return;
-        }
-        const self = this;
-        this._geolayer.setStyle(function (feature) {
-            return self._style(feature.properties);
-        });
-    }
-
     public AddNewElement(element) {
         this._newElements.push(element);
         console.log("Element added");
@@ -186,7 +176,15 @@ export class FilteredLayer {
             onEachFeature: function (feature, layer) {
                 let eventSource = self._storage.addOrGetElement(feature);
                 eventSource.addCallback(function () {
-                    self.updateStyle();
+                    if (layer.setIcon) {
+                        layer.setIcon(self._style(feature.properties).icon)
+                    } else {
+                        console.log("UPdating", layer);
+
+                        self._geolayer.setStyle(function (feature) {
+                            return self._style(feature.properties);
+                        });
+                    }
                 });