diff --git a/extensions/tags/src/TagState.php b/extensions/tags/src/TagState.php
index 38a7fde76..e697ccbb3 100644
--- a/extensions/tags/src/TagState.php
+++ b/extensions/tags/src/TagState.php
@@ -62,7 +62,7 @@ class TagState extends AbstractModel
      * @param Builder $query
      * @return Builder
      */
-    protected function setKeysForSaveQuery(Builder $query)
+    protected function setKeysForSaveQuery($query)
     {
         $query->where('tag_id', $this->tag_id)
               ->where('user_id', $this->user_id);