diff --git a/src/grelu/data/dataset.py b/src/grelu/data/dataset.py
index 2849090..56b0dfb 100755
--- a/src/grelu/data/dataset.py
+++ b/src/grelu/data/dataset.py
@@ -330,7 +330,7 @@ def __init__(
         # Get the labels
         if label_key is None:
             if scipy.sparse.issparse(adata.X):
-                labels = adata.X.A.T
+                labels = adata.X.toarray().T
             else:
                 labels = adata.X.T