diff --git a/Manifest.toml b/Manifest.toml
index 193446d..697edad 100644
--- a/Manifest.toml
+++ b/Manifest.toml
@@ -130,10 +130,8 @@ uuid = "56ddb016-857b-54e1-b83d-db4d58db5568"
 
 [[MLKernels]]
 deps = ["LinearAlgebra", "SpecialFunctions", "Statistics", "Test"]
-git-tree-sha1 = "456c6aa2928ae85064ef1d28b80d04fafed08986"
-repo-rev = "master"
-repo-url = "https://github.com/trthatcher/MLKernels.jl.git"
-uuid = "48eadcf2-8ff1-11e8-289d-f103432a0bb3"
+git-tree-sha1 = "d8ae6ac28480a25ddaae9ad9108371035ef9a346"
+uuid = "6899632a-1081-549c-8d71-752c8a25a7ba"
 version = "0.4.0"
 
 [[MLLabelUtils]]
diff --git a/Project.toml b/Project.toml
index e6fbdbc..f7f942f 100644
--- a/Project.toml
+++ b/Project.toml
@@ -8,7 +8,7 @@ Distributions = "31c24e10-a181-5473-b8eb-7969acd0382f"
 Ipopt = "b6b21f68-93f8-5de0-b562-5493be1d77c9"
 JuMP = "4076af6c-e467-56ae-b986-b466b2749572"
 LinearAlgebra = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e"
-MLKernels = "48eadcf2-8ff1-11e8-289d-f103432a0bb3"
+MLKernels = "6899632a-1081-549c-8d71-752c8a25a7ba"
 MLLabelUtils = "66a33bbf-0c2b-5fc8-a008-9da813334f0a"
 Memento = "f28f55f0-a522-5efc-85c2-fe41dfb9b2d9"
 Random = "9a3f8284-a2c9-5f02-9a11-845980a1fd5c"