diff --git a/tom_targets/templatetags/targets_extras.py b/tom_targets/templatetags/targets_extras.py
index 3b85a84d1..0f1effe49 100644
--- a/tom_targets/templatetags/targets_extras.py
+++ b/tom_targets/templatetags/targets_extras.py
@@ -70,14 +70,15 @@ def target_feature(target):
     return {'target': target}
 
 
-@register.inclusion_tag('tom_targets/partials/target_buttons.html')
-def target_buttons(target):
+@register.inclusion_tag('tom_targets/partials/target_buttons.html', takes_context=True)
+def target_buttons(context, target):
     """
     Displays the Update and Delete and Sharing buttons for a target.
     """
     sharing = getattr(settings, "DATA_SHARING", None)
-    return {'target': target,
-            'sharing': sharing}
+    context['target'] = target
+    context['sharing'] = sharing
+    return context
 
 
 @register.inclusion_tag('tom_targets/partials/target_data.html')