diff --git a/Project.toml b/Project.toml
index 285351c..f016c0d 100644
--- a/Project.toml
+++ b/Project.toml
@@ -1,7 +1,7 @@
 name = "DiagonalArrays"
 uuid = "74fd4be6-21e2-4f6f-823a-4360d37c7a77"
 authors = ["ITensor developers <support@itensor.org> and contributors"]
-version = "0.3.20"
+version = "0.3.21"
 
 [deps]
 ArrayLayouts = "4c555306-a7a7-4459-81d9-ec55ddd5c99a"
@@ -23,6 +23,6 @@ DerivableInterfaces = "0.5.5"
 FillArrays = "1.13.0"
 LinearAlgebra = "1.10.0"
 MapBroadcast = "0.1.10"
-MatrixAlgebraKit = "0.2, 0.3"
+MatrixAlgebraKit = "0.2, 0.3, 0.4"
 SparseArraysBase = "0.7.2"
 julia = "1.10"
diff --git a/test/Project.toml b/test/Project.toml
index 10b4543..5c86617 100644
--- a/test/Project.toml
+++ b/test/Project.toml
@@ -17,7 +17,7 @@ DerivableInterfaces = "0.5"
 DiagonalArrays = "0.3"
 FillArrays = "1"
 LinearAlgebra = "1"
-MatrixAlgebraKit = "0.2.5, 0.3"
+MatrixAlgebraKit = "0.2.5, 0.3, 0.4"
 SafeTestsets = "0.1"
 SparseArraysBase = "0.7"
 StableRNGs = "1"