Get current row when edit button clicked in WPF DataGrid?

in my DataGrid the last column is a TemplateColumn that has an Edit Button, the current implementation is that the user needs to check a checkbox beside the row and then click the edit button, obviously, this is nonsense, so isn't there a way to get the current row when I click the related edit button?