diff --git a/swaydb/src/main/scala/swaydb/Map.scala b/swaydb/src/main/scala/swaydb/Map.scala
index d656e9db3..90cde89e1 100644
--- a/swaydb/src/main/scala/swaydb/Map.scala
+++ b/swaydb/src/main/scala/swaydb/Map.scala
@@ -153,10 +153,10 @@ case class Map[K, V, F, T[_]](private[swaydb] val core: Core[T],
     function
   }
 
-  def applyFunction[B <: F with swaydb.Function[K, V]](key: K, function: B): T[IO.Done] =
+  def applyFunction(key: K, function: F with swaydb.Function[K, V]): T[IO.Done] =
     tag.point(core.function(key, function.id))
 
-  def applyFunction[B <: F with swaydb.Function[K, V]](from: K, to: K, function: B): T[IO.Done] =
+  def applyFunction(from: K, to: K, function: F with swaydb.Function[K, V]): T[IO.Done] =
     tag.point(core.function(from, to, function.id))
 
   def commit(prepare: Prepare[K, V]*): T[IO.Done] =
diff --git a/swaydb/src/main/scala/swaydb/Set.scala b/swaydb/src/main/scala/swaydb/Set.scala
index 68b545874..4a485fcdf 100644
--- a/swaydb/src/main/scala/swaydb/Set.scala
+++ b/swaydb/src/main/scala/swaydb/Set.scala
@@ -130,11 +130,11 @@ case class Set[A, F, T[_]](private val core: Core[T],
     function
   }
 
-  def applyFunction(from: A, to: A, functionId: A): T[IO.Done] =
-    tag.point(core.function(from, to, functionId))
+  def applyFunction(from: A, to: A, function: F with swaydb.Function.GetKey[A, Nothing]): T[IO.Done] =
+    tag.point(core.function(from, to, function.id))
 
-  def applyFunction(elem: A, function: A): T[IO.Done] =
-    tag.point(core.function(elem, function))
+  def applyFunction(elem: A, function: F with swaydb.Function.GetKey[A, Nothing]): T[IO.Done] =
+    tag.point(core.function(elem, function.id))
 
   def commit(prepare: Prepare[A, Nothing]*): T[IO.Done] =
     tag.point(core.put(prepare))