diff --git a/cogment_lab/cli/download_cogment.py b/cogment_lab/cli/download_cogment.py
index 83772c1..38d478f 100644
--- a/cogment_lab/cli/download_cogment.py
+++ b/cogment_lab/cli/download_cogment.py
@@ -31,10 +31,10 @@ class Arch(Enum):
 
 def get_current_arch() -> Arch:
     py_machine = platform.machine()
-    if py_machine in ["x86_64", "i686", "AMD64"]:
+    if py_machine in ["x86_64", "i686", "AMD64", "aarch64"]:
         return Arch.AMD64
 
-    if py_machine in ["arm64", "aarch64"]:
+    if py_machine in ["arm64"]:
         return Arch.ARM64
 
     raise RuntimeError(f"Unsupported architecture [{py_machine}]")