diff --git a/src/Traits/CompositeKeys.php b/src/Traits/CompositeKeys.php
index a26ee83..3564745 100644
--- a/src/Traits/CompositeKeys.php
+++ b/src/Traits/CompositeKeys.php
@@ -21,7 +21,7 @@ trait CompositeKeys {
 	 * @param  \Illuminate\Database\Eloquent\Builder  $query
 	 * @return \Illuminate\Database\Eloquent\Builder
 	 */
-	protected function setKeysForSaveQuery(Builder $query)
+	protected function setKeysForSaveQuery($query)
 	{
 		$keys = $this->getKeyName();