diff --git a/test/tool/tool.cpp b/test/tool/tool.cpp
index f189f0546c..14723343c7 100644
--- a/test/tool/tool.cpp
+++ b/test/tool/tool.cpp
@@ -500,13 +500,13 @@ void dump_context_array(hsa_queue_t* queue) {
           }
         }
       }
-
-      if (pthread_mutex_unlock(&mutex) != 0) {
-        perror("pthread_mutex_unlock");
-        abort();
-      }
-      if (done == false) sched_yield();
     }
+
+    if (pthread_mutex_unlock(&mutex) != 0) {
+      perror("pthread_mutex_unlock");
+      abort();
+    }
+    if (done == false) sched_yield();
   }
 }