diff --git a/public/gentrap/src/main/scala/nl/lumc/sasc/biopet/pipelines/gentrap/measures/Measurement.scala b/public/gentrap/src/main/scala/nl/lumc/sasc/biopet/pipelines/gentrap/measures/Measurement.scala
index 8e6a5bbd6e93e89628e92e4dbbbf86ddb9d30afa..095a01fdbb5ddf2024198dc01f11bd79b89c53de 100644
--- a/public/gentrap/src/main/scala/nl/lumc/sasc/biopet/pipelines/gentrap/measures/Measurement.scala
+++ b/public/gentrap/src/main/scala/nl/lumc/sasc/biopet/pipelines/gentrap/measures/Measurement.scala
@@ -47,11 +47,11 @@ trait Measurement extends SummaryQScript with Reference { qscript: QScript =>
     extraSummaryFiles += s"${name}_table" -> outputFile
   }
 
-  def addHeatmapJob(countTable: File, outputFile: File, name: String): Unit = {
+  def addHeatmapJob(countTable: File, outputFile: File, name: String, countType: Option[String] = None): Unit = {
     val job = new PlotHeatmap(qscript)
     job.input = countTable
     job.output = outputFile
-    job.countType = Some(name)
+    job.countType = countType
     add(job)
     extraSummaryFiles += s"${name}_heatmap" -> outputFile
   }