WPF behaviour to check/uncheck a checkbox in a list item

I have an application with several item controls (treeviews and others) which contain an item template with a checkbox inside. This checkbox checked state is bound to an IsChecked property of the item view model. This works correctly when clicking on the checkbox, but it's impossible to check/uncheck them with the keyboard (I believe this is due to the fact that the checkbox itself never gets the focus).

I like the solution proposed by DLeh in here: http://ift.tt/1zLAxgF but I would like an improvement: Instead of having the behaviour calling a command on the base view model (the vm which contains the list of items), I would like the behaviour to directly act on the IsChecked property of the item. My problem is that I don't know how to modify the behaviour or how to set up the binding on it, so that the behaviour can have access to the item's IsChecked property. So, instead of the following:

<DataGrid>
    <i:Interaction.Behaviors>
        <shared:ToggleSelectOnSpace ToggleSelectCommand="{Binding Data.ToggleSelectParticipantCommand, Source={StaticResource BindingProxy}}" />
    </i:Interaction.Behaviors>
...
</DataGrid>

I would have something like this:

<DataGrid>
    <i:Interaction.Behaviors>
        <shared:ToggleSelectOnSpace ItemsIsSelectedProperty="{Binding IsChecked}" />
    </i:Interaction.Behaviors>
...
</DataGrid>

Update I should add that my current implementation uses the PreviewKeyUp event in the itemscontrol and the following code behind implementation. The problem with this approach is that I have this code in many code behind files, so there is a lot of duplication. My goal is to replace this by a behaviour.

private void TreeView_OnPreviewKeyUp(object sender, KeyEventArgs e)
{
   if (e.Key == Key.Space)
   {
        var tree = (TreeView) sender;
        var item = tree.SelectedItem as IsSelectedViewModelBase;
        if (item != null)
        {
            item.IsSelected = !item.IsSelected;
            e.Handled = true;
        }
   }
}