diff --git a/DESCRIPTION b/DESCRIPTION
index f9230f6..482d9a8 100755
--- a/DESCRIPTION
+++ b/DESCRIPTION
@@ -3,7 +3,7 @@ Maintainer: Stefan Dentro <sd11@sanger.ac.uk>
 License: GPL-3
 Type: Package
 Title: DPClust - An R package for subclonal reconstruction of tumours using sequencing data
-Version: 2.2.7
+Version: 2.2.8
 Authors@R: c(person("David", "Wedge", role=c("aut"), email="dw9@sanger.ac.uk"),
     person("Peter", "Van Loo", role=c("aut")),
     person("Stefan","Dentro", email="sd11@sanger.ac.uk", role=c("aut", "cre")))
diff --git a/R/SampleMutations.R b/R/SampleMutations.R
index 248ed1d..6ae8d87 100755
--- a/R/SampleMutations.R
+++ b/R/SampleMutations.R
@@ -136,7 +136,7 @@ unsample_mutations = function(dataset, clustering_result) {
   
   # Not all assignment options return a full likelihood table
   if (!is.na(clustering_result$all.assignment.likelihoods)) {
-    all.assignment.likelihoods = clustering_result$all.assignment.likelihoods[dataset$most.similar.mut,]
+    all.assignment.likelihoods = clustering_result$all.assignment.likelihoods[dataset$most.similar.mut,,drop=F]
   } else {
     all.assignment.likelihoods = NA
   }