diff --git a/.github/workflows/python-package.yml b/.github/workflows/python-package.yml
index 350881a..d5797b5 100644
--- a/.github/workflows/python-package.yml
+++ b/.github/workflows/python-package.yml
@@ -27,8 +27,8 @@ jobs:
     - name: Install dependencies
       run: |
         python -m pip install --upgrade pip
+        python -m pip install torch==1.10.0
         python -m pip install pytest
-        if [ -f requirements.txt ]; then pip install -r requirements.txt; fi
     - name: Test with pytest
       run: |
         python setup.py pytest
diff --git a/invariant_point_attention/invariant_point_attention.py b/invariant_point_attention/invariant_point_attention.py
index c17cfd3..46eb19d 100644
--- a/invariant_point_attention/invariant_point_attention.py
+++ b/invariant_point_attention/invariant_point_attention.py
@@ -333,7 +333,7 @@ def forward(
             rotations = quaternion_to_matrix(quaternions)
 
             if self.detach_rotations:
-                rotations.detach_()
+                rotations = rotations.detach()
 
             x = block(
                 x,
diff --git a/setup.py b/setup.py
index f4dfe37..3cea4ab 100644
--- a/setup.py
+++ b/setup.py
@@ -3,7 +3,7 @@
 setup(
   name = 'invariant-point-attention',
   packages = find_packages(),
-  version = '0.2.1',
+  version = '0.2.2',
   license='MIT',
   description = 'Invariant Point Attention',
   long_description_content_type = 'text/markdown',