diff --git a/.github/workflows/integration-test.yml b/.github/workflows/integration-test.yml
index b149ab6..e135bbb 100644
--- a/.github/workflows/integration-test.yml
+++ b/.github/workflows/integration-test.yml
@@ -30,6 +30,7 @@ jobs:
           - {name: "scvi-tools", extras: ""}
           - {name: "pertpy", extras: "de"}
           - {name: "decoupler", extras: ""}
+          - {name: "SnapATAC2", extras: ""}
 
     defaults:
       run:
@@ -52,6 +53,7 @@ jobs:
       - uses: actions/checkout@v4
         with:
           repository: scverse/${{ matrix.package.name }}
+          ref: 'ig/unfiform_layout'
           fetch-depth: 0
           filter: blob:none
           path: ${{ matrix.package.name }}
@@ -63,6 +65,10 @@ jobs:
 
       - name: Install UV
         uses: hynek/setup-cached-uv@v2
+      
+      # For snapatac2
+      - name: Install rust-cache
+        uses: Swatinem/rust-cache@v2
 
       - name: Install AnnData and scverse package
         run: |
diff --git a/scvi-tools b/scvi-tools
new file mode 160000
index 0000000..c338ca0
--- /dev/null
+++ b/scvi-tools
@@ -0,0 +1 @@
+Subproject commit c338ca01b54dff7940ab31aed7b7e9c6c7180f1e