diff --git a/app/Handlers/Registrators/WidgetRegistrator.php b/app/Handlers/Registrators/WidgetRegistrator.php
index e70cf9ee..d3b68a3a 100644
--- a/app/Handlers/Registrators/WidgetRegistrator.php
+++ b/app/Handlers/Registrators/WidgetRegistrator.php
@@ -18,9 +18,11 @@ class WidgetRegistrator
     /**
      * Method used for registering new widgets
      *
-     * @param $classname
+     * @param      $classname
+     * @param null $templateId
+     * @param null $moduleId
      */
-    public function register($classname)
+    public function register($classname, $templateId = null, $moduleId = null)
     {
         $class = new $classname;
         if (!$class instanceof PublicWidget) {
@@ -36,6 +38,8 @@ class WidgetRegistrator
             WidgetType::create([
                                    'handler'     => $classname,
                                    'name'        => $widgetName,
+                                   'module_id'   => $moduleId,
+                                   'template_id' => $templateId,
                                    'description' => $widgetDescription
                                ]);