diff --git a/.kraken/unit_tests.py b/.kraken/unit_tests.py
index 952e7d6a..731cca10 100644
--- a/.kraken/unit_tests.py
+++ b/.kraken/unit_tests.py
@@ -36,9 +36,11 @@ def stage(ctx):
                 "cwd": "kraken/agent"
             }],
             "environments": [{
-                "system": "krakenci/ubuntu:20.04",
-                "agents_group": "docker",
-                "executor": "docker",
+                #"system": "krakenci/ubuntu:20.04",
+                #"agents_group": "docker",
+                #"executor": "docker",
+                "system": "Canonical:0001-com-ubuntu-server-focal:20_04-lts:20.04.202109080",
+                "agents_group": "azure-vm",
                 "config": "default"
             }]
         }, {