diff --git a/handlers/main.yml b/handlers/main.yml
index 151f3f4..1c1dbd8 100644
--- a/handlers/main.yml
+++ b/handlers/main.yml
@@ -1,6 +1,6 @@
 ---
 - name: Restart Telegraf
-  systemd:
+  service:
     name: telegraf
     state: restarted
     daemon_reload: yes
diff --git a/tasks/setup.yml b/tasks/setup.yml
index 0dba037..7cf3e99 100644
--- a/tasks/setup.yml
+++ b/tasks/setup.yml
@@ -52,7 +52,7 @@
       notify: __telegraf_restart
 
     - name: Ensure Telegraf is up and running
-      systemd:
+      service:
         name: telegraf
         daemon_reload: yes
         enabled: yes