diff --git a/nixos/tests/all-tests.nix b/nixos/tests/all-tests.nix
index 8fd22251b90a..e4309a98f93e 100644
--- a/nixos/tests/all-tests.nix
+++ b/nixos/tests/all-tests.nix
@@ -1163,7 +1163,7 @@ in {
   tayga = handleTest ./tayga.nix {};
   technitium-dns-server = handleTest ./technitium-dns-server.nix {};
   teeworlds = handleTest ./teeworlds.nix {};
-  telegraf = handleTest ./telegraf.nix {};
+  telegraf = runTest ./telegraf.nix;
   teleport = handleTest ./teleport.nix {};
   teleports = runTest ./teleports.nix;
   thelounge = handleTest ./thelounge.nix {};
diff --git a/nixos/tests/telegraf.nix b/nixos/tests/telegraf.nix
index 93d4c85b3e8d..e09b1848882f 100644
--- a/nixos/tests/telegraf.nix
+++ b/nixos/tests/telegraf.nix
@@ -1,47 +1,45 @@
-import ./make-test-python.nix (
-  { pkgs, ... }:
-  {
-    name = "telegraf";
-    meta = with pkgs.lib.maintainers; {
-      maintainers = [ mic92 ];
+{ pkgs, ... }:
+{
+  name = "telegraf";
+  meta = with pkgs.lib.maintainers; {
+    maintainers = [ mic92 ];
+  };
+
+  nodes.machine =
+    { ... }:
+    {
+      services.telegraf.enable = true;
+      services.telegraf.environmentFiles = [
+        (pkgs.writeText "secrets" ''
+          SECRET=example
+        '')
+      ];
+      services.telegraf.extraConfig = {
+        agent.interval = "1s";
+        agent.flush_interval = "1s";
+        inputs.exec = {
+          commands = [
+            "${pkgs.runtimeShell} -c 'echo $SECRET,tag=a i=42i'"
+          ];
+          timeout = "5s";
+          data_format = "influx";
+        };
+        inputs.ping = {
+          urls = [ "127.0.0.1" ];
+          count = 4;
+          interval = "10s";
+          timeout = 1.0;
+        };
+        outputs.file.files = [ "/tmp/metrics.out" ];
+        outputs.file.data_format = "influx";
+      };
     };
 
-    nodes.machine =
-      { ... }:
-      {
-        services.telegraf.enable = true;
-        services.telegraf.environmentFiles = [
-          (pkgs.writeText "secrets" ''
-            SECRET=example
-          '')
-        ];
-        services.telegraf.extraConfig = {
-          agent.interval = "1s";
-          agent.flush_interval = "1s";
-          inputs.exec = {
-            commands = [
-              "${pkgs.runtimeShell} -c 'echo $SECRET,tag=a i=42i'"
-            ];
-            timeout = "5s";
-            data_format = "influx";
-          };
-          inputs.ping = {
-            urls = [ "127.0.0.1" ];
-            count = 4;
-            interval = "10s";
-            timeout = 1.0;
-          };
-          outputs.file.files = [ "/tmp/metrics.out" ];
-          outputs.file.data_format = "influx";
-        };
-      };
+  testScript = ''
+    start_all()
 
-    testScript = ''
-      start_all()
-
-      machine.wait_for_unit("telegraf.service")
-      machine.wait_until_succeeds("grep -q example /tmp/metrics.out")
-      machine.wait_until_succeeds("grep -q ping /tmp/metrics.out")
-    '';
-  }
-)
+    machine.wait_for_unit("telegraf.service")
+    machine.wait_until_succeeds("grep -q example /tmp/metrics.out")
+    machine.wait_until_succeeds("grep -q ping /tmp/metrics.out")
+  '';
+}