diff --git a/+diff/mmd.m b/+diff/mmd.m
index 9c15dd3..d7ff872 100644
--- a/+diff/mmd.m
+++ b/+diff/mmd.m
@@ -57,9 +57,9 @@
    sigma = par.Results.sigma;
 end
 
-K = utils.rbf(sigma,x);
-L = utils.rbf(sigma,y);
-KL = utils.rbf(sigma,x,y);
+K = utils.rbf(x,x);
+L = utils.rbf(y,y);
+KL = utils.rbf(x,y,sigma);
 K = utils.zerodiag(K);
 L = utils.zerodiag(L);