diff --git a/src/main/java/qupath/ext/instanseg/core/PytorchManager.java b/src/main/java/qupath/ext/instanseg/core/PytorchManager.java
index f0ba496..53b5eb4 100644
--- a/src/main/java/qupath/ext/instanseg/core/PytorchManager.java
+++ b/src/main/java/qupath/ext/instanseg/core/PytorchManager.java
@@ -50,7 +50,7 @@ public static Collection<String> getAvailableDevices() {
                 availableDevices.add(name);
             }
             // If we could use MPS, but don't have it already, add it
-            if (GeneralTools.isMac() && "aarch64".equals(System.getProperty("os.arch"))) {
+            if (GeneralTools.isMac()) {
                 availableDevices.add("mps");
             }
             // CPU should always be available