diff --git a/hipamd/src/hip_graph_internal.cpp b/hipamd/src/hip_graph_internal.cpp
index 3c118bc198..847c72f9d0 100644
--- a/hipamd/src/hip_graph_internal.cpp
+++ b/hipamd/src/hip_graph_internal.cpp
@@ -574,7 +574,7 @@ hipError_t hipGraphExec::Run(hipStream_t stream) {
   if (hip::getStream(stream) == nullptr) {
     return hipErrorInvalidResourceHandle;
   }
-  if (flags_ == hipGraphInstantiateFlagAutoFreeOnLaunch) {
+  if (flags_ & hipGraphInstantiateFlagAutoFreeOnLaunch) {
     if (!levelOrder_.empty()) {
       levelOrder_[0]->GetParentGraph()->FreeAllMemory();
     }