diff --git a/Dockerfile b/Dockerfile
index 777f9eb..bc7207a 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,4 +1,4 @@
-FROM waggle/plugin-base:1.1.1-ml-cuda10.2-l4t
+FROM waggle/plugin-base:1.1.1-ml
 
 COPY requirements.txt /app/
 RUN pip3 install --no-cache-dir -r /app/requirements.txt
diff --git a/requirements.txt b/requirements.txt
index 8acca23..dfe1c32 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,2 +1,2 @@
 git+https://github.com/sagecontinuum/sage-cli.git
-https://github.com/waggle-sensor/pywaggle/archive/0.43.2.zip
+waggle[all] @ https://github.com/waggle-sensor/pywaggle/releases/download/0.46.2/waggle-0.46.2-py3-none-any.whl
diff --git a/sage.yaml b/sage.yaml
index f5abd0d..d2b930e 100644
--- a/sage.yaml
+++ b/sage.yaml
@@ -1,6 +1,6 @@
 name: "cloudcover-unet"
 description: "U-Net Cloud Coverage Estimator"
-version : "0.0.0"
+version : "0.0.1"
 namespace: "sage"
 authors: "Seongha Park <seongha.park@anl.gov>"
 license: "MIT license"
@@ -9,6 +9,7 @@ keywords: "machine learning, U-Net, image segmentation, cloud coverage, ground-b
 source:
   architectures :
     - "linux/arm64"
+    - "linux/amd64"
   url: "https://github.com/nconrad/plugin-cloudcover-unet.git"
   build_args:
     SAGE_STORE_URL: "https://osn.sagecontinuum.org"