diff --git a/omegafold/pipeline.py b/omegafold/pipeline.py
index 146c39e..9512783 100644
--- a/omegafold/pipeline.py
+++ b/omegafold/pipeline.py
@@ -43,7 +43,7 @@
 
 try:
     from torch.backends import mps  # Compatibility with earlier versions
-except IndexError:
+except ImportError:
     mps = None