diff --git a/DESCRIPTION b/DESCRIPTION
index 18fb1581..7ae386ff 100644
--- a/DESCRIPTION
+++ b/DESCRIPTION
@@ -39,6 +39,7 @@ Imports:
     ggridges,
     ggpubr,
     htmlwidgets,
+    iheatmap,
     lme4,
     lmerTest,
     MASS,
diff --git a/man/plot_heatmap_cor_iheatmap.Rd b/man/plot_heatmap_cor_iheatmap.Rd
index 47077c3e..60155d69 100644
--- a/man/plot_heatmap_cor_iheatmap.Rd
+++ b/man/plot_heatmap_cor_iheatmap.Rd
@@ -22,10 +22,9 @@ config <- istar$config
 analysis <- istar$data
 
 pheat_map <- plot_heatmap_cor_iheatmap( analysis, config )
-debug(plot_heatmap_cor_iheatmap)
-stopifnot("pheatmap" \%in\% class(pheat_map))
+stopifnot("IheatmapHorizontal" \%in\% class(pheat_map))
 pheat_map <- plot_heatmap_cor_iheatmap( analysis, config, R2 = TRUE )
-stopifnot("pheatmap" \%in\% class(pheat_map))
+stopifnot("IheatmapHorizontal" \%in\% class(pheat_map))
 
 }
 \seealso{