No you were not wrong.
That said only you know if you were impacted by the death and in the correct state of mind to dive. I have seen horrific body mutilations that have impacted me days after the event - in those types of situations I would not dive and would caution anyone to seek help