diff --git a/flake.nix b/flake.nix
index f26aff2..27065a5 100644
--- a/flake.nix
+++ b/flake.nix
@@ -33,14 +33,14 @@
             "cmd/pruner"
           ];
         };
-        ws = pkgs.dockerTools.buildImage {
+        container = pkgs.dockerTools.buildImage {
           name = "ps2.live/saerro/ws";
           tag = "latest";
           copyToRoot = [ 
             pkgs.dockerTools.caCertificates
           ];
           config = {
-            Cmd = [ "${isolateBinary saerro "ws"}/bin/ws" ];
+            Cmd = [ "${saerro}/bin/ws" ];
           };
         };
         pruner = pkgs.dockerTools.buildImage {
@@ -50,7 +50,7 @@
             pkgs.dockerTools.caCertificates
           ];
           config = {
-            Cmd = [ "${isolateBinary saerro "pruner"}/bin/pruner" ];
+            Cmd = [ "${saerro}/bin/pruner" ];
           };
         };
       };