diff --git a/Docker/Dockerfile b/Docker/Dockerfile
index c4d7a2777..402c081de 100644
--- a/Docker/Dockerfile
+++ b/Docker/Dockerfile
@@ -3,7 +3,7 @@ WORKDIR /app
 COPY . .
 RUN apk add --no-cache --virtual .build-deps make g++ \
     && apk add --no-cache python3 \
-    && if [ "$(uname -m)" = "x86_64" ]; then node setup noShortcuts tensorflow; else node setup noShortcuts; fi \
+    && npm ci \
     && mkdir -p ./Platform/My-Data-Storage ./Platform/My-Log-Files ./Platform/My-Workspaces \
     && addgroup superalgos \
     && adduser --disabled-password --no-create-home --ingroup superalgos superalgos \
diff --git a/package.json b/package.json
index 95eb16937..724210a3a 100644
--- a/package.json
+++ b/package.json
@@ -35,7 +35,9 @@
     "twitter": "^1.7.1",
     "util": "^0.12.4",
     "web3": "^1.6.0",
-    "ws": "^8.2.3",
+    "ws": "^8.2.3"
+  },
+  "optionalDependencies": {
     "@tensorflow/tfjs-node": "^2.8.6"
   },
   "devDependencies": {