Federal agent question?

Lets say a federal agent, such as a border patrol agent or dea or even an atf agent gets shot or killed by a criminal. Who gets put in charge of the investigation? The agency they work for? Or the FBI? I think its the FBI, it doesn't matter what agency they were from since they were feds.