diff --git a/hipamd/src/hip_graph.cpp b/hipamd/src/hip_graph.cpp
index 3dc9ee7614..9ccf2c2b47 100644
--- a/hipamd/src/hip_graph.cpp
+++ b/hipamd/src/hip_graph.cpp
@@ -1292,7 +1292,8 @@ hipError_t hipGraphGetEdges(hipGraph_t graph, hipGraphNode_t* from, hipGraphNode
                             size_t* numEdges) {
   HIP_INIT_API(hipGraphGetEdges, graph, from, to, numEdges);
   if (graph == nullptr || numEdges == nullptr ||
-      from == nullptr || to == nullptr) {
+      (from == nullptr && to != nullptr) ||
+      (to == nullptr && from != nullptr)) {
     HIP_RETURN(hipErrorInvalidValue);
   }
   const std::vector<std::pair<Node, Node>> edges = graph->GetEdges();