Nobody ever "punished" Vito for the failed hit because he was probably the most feared mobster in the country at that point. Once he went to jail he lost a lot of power, even within his own family, and there was not a lot he could do as far as getting back at Gambino, who had consolidated his own power after Vito's imprisonment.