diff --git a/key_schedule.go b/key_schedule.go
index 9c76ebe..e8ee9ce 100644
--- a/key_schedule.go
+++ b/key_schedule.go
@@ -59,7 +59,7 @@ type keySharePrivateKeys struct {
 
 // kyberDecapsulate implements decapsulation according to Kyber Round 3.
 func kyberDecapsulate(dk *mlkem768.DecapsulationKey, c []byte) ([]byte, error) {
-	K, err := mlkem768.Decapsulate(dk, c)
+	K, err := dk.Decapsulate(c)
 	if err != nil {
 		return nil, err
 	}