diff --git a/src/Controller/Admin.php b/src/Controller/Admin.php
index 0a8314c..fe0a1b9 100644
--- a/src/Controller/Admin.php
+++ b/src/Controller/Admin.php
@@ -130,9 +130,8 @@ public function get_templates(\Box_App $app):string
      * Enables the QEMU Template
      *
      * @param \Box_App $app
-     * @return void
      */
-    public function enable_template(\Box_App $app, $id): void
+    public function enable_template(\Box_App $app, $id)
     {
         $api = $this->di['api_admin'];
         $api->Serviceproxmox_vm_config_template_enable(array('id' => $id));
@@ -143,9 +142,8 @@ public function enable_template(\Box_App $app, $id): void
      * Disables the QEMU Template
      *
      * @param \Box_App $app
-     * @return void
      */
-    public function disable_template(\Box_App $app, $id): void
+    public function disable_template(\Box_App $app, $id)
     {
         $api = $this->di['api_admin'];
         $api->Serviceproxmox_vm_config_template_disable(array('id' => $id));
@@ -265,9 +263,8 @@ public function get_vm_config_template(\Box_App $app, $id): string
      * Renders the admin area settings page
      * 
      * @param \Box_App $app
-     * @return void
      */
-    public function start_backup(\Box_App $app):void
+    public function start_backup(\Box_App $app)
     {
 
         $api = $this->di['api_admin'];
diff --git a/src/Service.php b/src/Service.php
index 3ad21a6..97d25eb 100644
--- a/src/Service.php
+++ b/src/Service.php
@@ -38,7 +38,10 @@ public function setDi(\Pimple\Container|null $di): void
 	{
 		$this->di = $di;
 	}
-
+    public function getDi(): ?\Pimple\Container
+    {
+        return $this->di;
+    }
 	use ProxmoxAuthentication;
 	use ProxmoxServer;
 	use ProxmoxVM;