diff --git a/.github/workflows/r.yml b/.github/workflows/r.yml
index a25afe1..a35428b 100644
--- a/.github/workflows/r.yml
+++ b/.github/workflows/r.yml
@@ -19,10 +19,8 @@ jobs:
       max-parallel: 5
       matrix:
         config:
-          - { r-version: 4.0, os: ubuntu-latest, python-version: 3.8 }
-          - { r-version: 4.1, os: ubuntu-latest, python-version: 3.8 }
-          - { r-version: 4.0, os: macos-latest, python-version: 3.8 }
-          - { r-version: 4.1, os: macos-latest, python-version: 3.8 }
+          - { r-version: release, os: ubuntu-latest}
+          - { r-version: release, os: macos-latest}
     runs-on: ${{ matrix.config.os }}
     env:
       R_REMOTES_NO_ERRORS_FROM_WARNINGS: true
diff --git a/R/plot_cpdb_heatmap.R b/R/plot_cpdb_heatmap.R
index 015dd0a..e6e50aa 100644
--- a/R/plot_cpdb_heatmap.R
+++ b/R/plot_cpdb_heatmap.R
@@ -110,7 +110,7 @@ plot_cpdb_heatmap <- function(scdata, idents, pvals, log1p_transform = FALSE, sh
       main = main, treeheight_row = treeheight_row, family = family, color = col.heatmap,
       treeheight_col = treeheight_col, ...)
     if (return_tables) {
-      return(list(count_network = count_matrix, interaction_count = all_sum))
+      return(list(count_network = count_mat, interaction_count = all_sum))
     } else {
       return(p)
     }
@@ -119,4 +119,4 @@ plot_cpdb_heatmap <- function(scdata, idents, pvals, log1p_transform = FALSE, sh
     stop("There are no significant results using p-value of: ", alpha, call. = FALSE)
   }
 
-}
\ No newline at end of file
+}