diff --git a/docs/index.rst b/docs/index.rst
index 0753228..8f6868d 100644
--- a/docs/index.rst
+++ b/docs/index.rst
@@ -35,7 +35,7 @@ Welcome to spapros's documentation!
 |logo|
 
 
-**Spapros** provides a pipeline for probe set selection and evaluation for targeted spatial transcriptomics data.
+**Spapros** is a python package that provides a pipeline for probe set selection and evaluation for targeted spatial transcriptomics data.
 
 Key Features
 -------------
diff --git a/spapros/selection/selection_procedure.py b/spapros/selection/selection_procedure.py
index 78000d0..62d829c 100644
--- a/spapros/selection/selection_procedure.py
+++ b/spapros/selection/selection_procedure.py
@@ -1679,7 +1679,7 @@ def plot_coexpression(
 
     def plot_clf_genes(
         self,
-        basis: int = "X_umap",
+        basis: str = "X_umap",
         celltypes: Optional[List[str]] = None,
         till_rank: Optional[int] = 1,
         importance_th: Optional[float] = None,