diff --git a/src/pdr.c b/src/pdr.c
index 4acdbcf..4cb2bf8 100644
--- a/src/pdr.c
+++ b/src/pdr.c
@@ -986,6 +986,10 @@
 				  pldm_entity entity, pldm_entity_node **node)
 {
 	assert(tree != NULL);
+	if (!tree || !node) {
+		return;
+	}
+
 	find_entity_ref_in_tree(tree->root, entity, node);
 }