diff --git a/plugins/tps/reklama/components/Advertisement.php b/plugins/tps/reklama/components/Advertisement.php
index 0379a7efd..bac062bc5 100644
--- a/plugins/tps/reklama/components/Advertisement.php
+++ b/plugins/tps/reklama/components/Advertisement.php
@@ -11,7 +11,7 @@ use Tps\Reklama\Models\Statistika;
 
 class Advertisement extends ComponentBase
 {
-    public $group;
+    public $group, $css_class;
     public function componentDetails()
     {
         return [
@@ -68,9 +68,9 @@ class Advertisement extends ComponentBase
         ];
     }
 
-    public function onRun()
+    public function onRender()
     {
-        parent::onRun(); // TODO: Change the autogenerated stub
+        $this->css_class = $this->property('css_class');
     }
 
     public function onDisplay(){
diff --git a/plugins/tps/reklama/components/advertisement/single.htm b/plugins/tps/reklama/components/advertisement/single.htm
index 029ae11cd..0dfea11c3 100644
--- a/plugins/tps/reklama/components/advertisement/single.htm
+++ b/plugins/tps/reklama/components/advertisement/single.htm
@@ -1,11 +1,3 @@
-==
-<?php
-function onRender() {
-    $css_class = $this->property('css_class');
-}
-?>
-
-==
 
 {% set reklama = group.adds.first %}
 <a