diff --git a/pxt.json b/pxt.json
index 4424ee6..6a1d1ec 100644
--- a/pxt.json
+++ b/pxt.json
@@ -4,7 +4,7 @@
     "description": "Machine learning extension to support micro:bit CreateAI",
     "dependencies": {
         "core": "*",
-        "machine-learning-runner": "github:microbit-foundation/pxt-microbit-ml-runner#v0.4.0"
+        "machine-learning-runner": "github:microbit-foundation/pxt-microbit-ml-runner#v0.4.4"
     },
     "files": [
         "README.md",