diff --git a/db/compaction_picker.cc b/db/compaction_picker.cc
index dad59a0a3..91b9a2b5a 100644
--- a/db/compaction_picker.cc
+++ b/db/compaction_picker.cc
@@ -141,7 +141,8 @@ CompactionPicker::~CompactionPicker() {}
 
 // Delete this compaction from the list of running compactions.
 void CompactionPicker::ReleaseCompactionFiles(Compaction* c, Status status) {
-  if (c->start_level() == 0) {
+  if (c->start_level() == 0 ||
+      ioptions_.compaction_style == kCompactionStyleUniversal) {
     level0_compactions_in_progress_.erase(c);
   }
   if (!status.ok()) {