diff --git a/src/Controller/AJAXLinkController.php b/src/Controller/AJAXLinkController.php
index 754b506..3d4dd77 100644
--- a/src/Controller/AJAXLinkController.php
+++ b/src/Controller/AJAXLinkController.php
@@ -38,7 +38,6 @@ class AJAXLinkController extends ControllerBase {
 
     $flagging = \Drupal::service('flag')->flag($flag_id, $entity_id);
 
-    $flag = $flagging->getFlag();
     $entity = $flagging->getFlaggable();
 
     return $this->generateResponse('unflag', $flag, $entity);
@@ -66,7 +65,6 @@ class AJAXLinkController extends ControllerBase {
     $flag_service = \Drupal::service('flag');
     $flag_service->unflag($flag_id, $entity_id);
 
-    $flag = $flag_service->getFlagById($flag_id);
     $entity = $flag_service->getFlaggableById($flag, $entity_id);
 
     return $this->generateResponse('flag', $flag, $entity);