diff --git a/snips/Dockerfile b/snips/Dockerfile
index e0dc9ea..fc30e0e 100644
--- a/snips/Dockerfile
+++ b/snips/Dockerfile
@@ -1,5 +1,6 @@
 ARG BUILD_FROM
-FROM $BUILD_FROM
+ARG BUILD_VERSION
+FROM ${BUILD_FROM}-${BUILD_VERSION}
 
 # Copy data
 COPY run.sh /
diff --git a/snips/build.json b/snips/build.json
index c5c2ec1..30d2a89 100644
--- a/snips/build.json
+++ b/snips/build.json
@@ -1,6 +1,6 @@
 {
   "build_from": {
-    "armhf": "snipsdocker/platform:arm-0.4.9",
-    "amd64": "snipsdocker/platform:x86-0.4.9"
+    "armhf": "snipsdocker/platform:arm",
+    "amd64": "snipsdocker/platform:x86"
   }
 }
diff --git a/snips/config.json b/snips/config.json
index 2cd14d6..7c2da68 100644
--- a/snips/config.json
+++ b/snips/config.json
@@ -1,6 +1,6 @@
 {
   "name": "Snips.AI",
-  "version": "0.4.9",
+  "version": "0.5.4",
   "slug": "snips",
   "description": "Local voice control platform",
   "url": "https://home-assistant.io/addons/snips/",