diff --git a/tests/testthat/test-boxplot.R b/tests/testthat/test-boxplot.R
index 66d8538..3c71d51 100644
--- a/tests/testthat/test-boxplot.R
+++ b/tests/testthat/test-boxplot.R
@@ -1,54 +1,55 @@
 test_that("boxplot for ranked single-task data set has no title", {
   data <- rbind(
     data.frame(algo="A1", value=0.8, case="C1"),
     data.frame(algo="A2", value=0.6, case="C1"),
     data.frame(algo="A3", value=0.4, case="C1"),
     data.frame(algo="A1", value=0.2, case="C2"),
     data.frame(algo="A2", value=0.1, case="C2"),
     data.frame(algo="A3", value=0.0, case="C2"))
 
   challenge <- as.challenge(data, taskName="T1", algorithm="algo", case="case", value="value", smallBetter=FALSE)
 
   ranking <- challenge%>%aggregateThenRank(FUN=median, ties.method="min")
 
-  actualPlot <- boxplot(ranking)[[1]]
+  actualPlot <- boxplot(ranking)
   expect_is(actualPlot, "ggplot")
   expect_equal(actualPlot$labels$title, NULL)
 })
 
 test_that("boxplots for ranked multi-task data set have titles", {
   dataTask1 <- cbind(task="T1",
                      rbind(
                         data.frame(algo="A1", value=0.8, case="C1"),
                         data.frame(algo="A2", value=0.6, case="C1"),
                         data.frame(algo="A3", value=0.4, case="C1"),
                         data.frame(algo="A1", value=0.2, case="C2"),
                         data.frame(algo="A2", value=0.1, case="C2"),
                         data.frame(algo="A3", value=0.0, case="C2")
                       ))
    dataTask2 <- cbind(task="T2",
                       rbind(
                         data.frame(algo="A1", value=0.2, case="C1"),
                         data.frame(algo="A2", value=0.3, case="C1"),
                         data.frame(algo="A3", value=0.4, case="C1"),
                         data.frame(algo="A1", value=0.7, case="C2"),
                         data.frame(algo="A2", value=0.8, case="C2"),
                         data.frame(algo="A3", value=0.9, case="C2")
                       ))
 
    data <- rbind(dataTask1, dataTask2)
 
    challenge <- as.challenge(data, by="task", algorithm="algo", case="case", value="value", smallBetter=FALSE)
 
    ranking <- challenge%>%aggregateThenRank(FUN=median, ties.method="min")
 
    actualPlots <- boxplot(ranking)
    actualPlotTask1 <- actualPlots[[1]]
    actualPlotTask2 <- actualPlots[[2]]
 
    expect_is(actualPlotTask1, "ggplot")
    expect_equal(actualPlotTask1$labels$title, "T1")
 
    expect_is(actualPlotTask2, "ggplot")
    expect_equal(actualPlotTask2$labels$title, "T2")
 })
+
diff --git a/tests/testthat/test-rankingHeatmap.R b/tests/testthat/test-rankingHeatmap.R
index 5c77967..33a0733 100644
--- a/tests/testthat/test-rankingHeatmap.R
+++ b/tests/testthat/test-rankingHeatmap.R
@@ -1,54 +1,54 @@
 test_that("ranking heatmap for single-task data set has no title", {
   data <- rbind(
     data.frame(algo="A1", value=0.8, case="C1"),
     data.frame(algo="A2", value=0.6, case="C1"),
     data.frame(algo="A3", value=0.4, case="C1"),
     data.frame(algo="A1", value=0.2, case="C2"),
     data.frame(algo="A2", value=0.1, case="C2"),
     data.frame(algo="A3", value=0.0, case="C2"))
 
   challenge <- as.challenge(data, taskName="T1", algorithm="algo", case="case", value="value", smallBetter=FALSE)
 
   ranking <- challenge%>%aggregateThenRank(FUN=median, ties.method="min")
 
-  actualPlot <- rankingHeatmap(ranking)[[1]]
+  actualPlot <- rankingHeatmap(ranking)
   expect_is(actualPlot, "ggplot")
   expect_equal(actualPlot$labels$title, NULL)
 })
 
 test_that("ranking heatmap for multi-task data set have titles", {
   dataTask1 <- cbind(task="T1",
                      rbind(
                         data.frame(algo="A1", value=0.8, case="C1"),
                         data.frame(algo="A2", value=0.6, case="C1"),
                         data.frame(algo="A3", value=0.4, case="C1"),
                         data.frame(algo="A1", value=0.2, case="C2"),
                         data.frame(algo="A2", value=0.1, case="C2"),
                         data.frame(algo="A3", value=0.0, case="C2")
                       ))
    dataTask2 <- cbind(task="T2",
                       rbind(
                         data.frame(algo="A1", value=0.2, case="C1"),
                         data.frame(algo="A2", value=0.3, case="C1"),
                         data.frame(algo="A3", value=0.4, case="C1"),
                         data.frame(algo="A1", value=0.7, case="C2"),
                         data.frame(algo="A2", value=0.8, case="C2"),
                         data.frame(algo="A3", value=0.9, case="C2")
                       ))
 
    data <- rbind(dataTask1, dataTask2)
 
    challenge <- as.challenge(data, by="task", algorithm="algo", case="case", value="value", smallBetter=FALSE)
 
    ranking <- challenge%>%aggregateThenRank(FUN=median, ties.method="min")
 
    actualPlots <- rankingHeatmap(ranking)
    actualPlotTask1 <- actualPlots[[1]]
    actualPlotTask2 <- actualPlots[[2]]
 
    expect_is(actualPlotTask1, "ggplot")
    expect_equal(actualPlotTask1$labels$title, "T1")
 
    expect_is(actualPlotTask2, "ggplot")
    expect_equal(actualPlotTask2$labels$title, "T2")
 })
diff --git a/tests/testthat/test-significanceMap.R b/tests/testthat/test-significanceMap.R
index 6f04755..6f59f32 100644
--- a/tests/testthat/test-significanceMap.R
+++ b/tests/testthat/test-significanceMap.R
@@ -1,54 +1,54 @@
 test_that("significance map for single-task data set has no title", {
   data <- rbind(
     data.frame(algo="A1", value=0.8, case="C1"),
     data.frame(algo="A2", value=0.6, case="C1"),
     data.frame(algo="A3", value=0.4, case="C1"),
     data.frame(algo="A1", value=0.2, case="C2"),
     data.frame(algo="A2", value=0.1, case="C2"),
     data.frame(algo="A3", value=0.0, case="C2"))
 
   challenge <- as.challenge(data, taskName="T1", algorithm="algo", case="case", value="value", smallBetter=FALSE)
 
   ranking <- challenge%>%aggregateThenRank(FUN=median, ties.method="min")
 
-  actualPlot <- significanceMap(ranking)[[1]]
+  actualPlot <- significanceMap(ranking)
   expect_is(actualPlot, "ggplot")
   expect_equal(actualPlot$labels$title, NULL)
 })
 
 test_that("significance map for multi-task data set have titles", {
   dataTask1 <- cbind(task="T1",
                      rbind(
                        data.frame(algo="A1", value=0.8, case="C1"),
                        data.frame(algo="A2", value=0.6, case="C1"),
                        data.frame(algo="A3", value=0.4, case="C1"),
                        data.frame(algo="A1", value=0.2, case="C2"),
                        data.frame(algo="A2", value=0.1, case="C2"),
                        data.frame(algo="A3", value=0.0, case="C2")
                      ))
   dataTask2 <- cbind(task="T2",
                      rbind(
                        data.frame(algo="A1", value=0.2, case="C1"),
                        data.frame(algo="A2", value=0.3, case="C1"),
                        data.frame(algo="A3", value=0.4, case="C1"),
                        data.frame(algo="A1", value=0.7, case="C2"),
                        data.frame(algo="A2", value=0.8, case="C2"),
                        data.frame(algo="A3", value=0.9, case="C2")
                      ))
 
   data <- rbind(dataTask1, dataTask2)
 
   challenge <- as.challenge(data, by="task", algorithm="algo", case="case", value="value", smallBetter=FALSE)
 
   ranking <- challenge%>%aggregateThenRank(FUN=median, ties.method="min")
 
   actualPlots <- significanceMap(ranking)
   actualPlotTask1 <- actualPlots[[1]]
   actualPlotTask2 <- actualPlots[[2]]
 
   expect_is(actualPlotTask1, "ggplot")
   expect_equal(actualPlotTask1$labels$title, "T1")
 
   expect_is(actualPlotTask2, "ggplot")
   expect_equal(actualPlotTask2$labels$title, "T2")
 })