I could be wrong, but I don’t think that is the only reason for the minhag.
It’s not.
Another reason brought is based on the gemara which says that someone who is careful in this mitzvah will have sons and sons-in-law who are talmidei chachomim, so when a child is born, a candle is added as a z’chus.