diff --git a/Project.toml b/Project.toml
index 50d3d73..0909e7e 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.16"
+version = "0.3.17"
 
 [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"
+MatrixAlgebraKit = "0.2, 0.3"
 SparseArraysBase = "0.7.2"
 julia = "1.10"
diff --git a/ext/DiagonalArraysMatrixAlgebraKitExt/DiagonalArraysMatrixAlgebraKitExt.jl b/ext/DiagonalArraysMatrixAlgebraKitExt/DiagonalArraysMatrixAlgebraKitExt.jl
index fd90124..3f94c39 100644
--- a/ext/DiagonalArraysMatrixAlgebraKitExt/DiagonalArraysMatrixAlgebraKitExt.jl
+++ b/ext/DiagonalArraysMatrixAlgebraKitExt/DiagonalArraysMatrixAlgebraKitExt.jl
@@ -125,7 +125,9 @@ for f in [
 ]
   for Alg in [:ScaledDeltaAlgorithm, :DeltaAlgorithm]
     @eval begin
-      function MatrixAlgebraKit.initialize_output(::typeof($f), a, alg::$Alg)
+      function MatrixAlgebraKit.initialize_output(
+        ::typeof($f), a::AbstractMatrix, alg::$Alg
+      )
         return nothing
       end
     end
@@ -148,13 +150,17 @@ for f in [
   :svd_vals!,
 ]
   @eval begin
-    function MatrixAlgebraKit.check_input(::typeof($f), a, F, alg::DeltaAlgorithm)
+    function MatrixAlgebraKit.check_input(
+      ::typeof($f), a::AbstractMatrix, F, alg::DeltaAlgorithm
+    )
       @assert size(a, 1) == size(a, 2)
       @assert isdiag(a)
       @assert all(isone, diagview(a))
       return nothing
     end
-    function MatrixAlgebraKit.check_input(::typeof($f), a, F, alg::ScaledDeltaAlgorithm)
+    function MatrixAlgebraKit.check_input(
+      ::typeof($f), a::AbstractMatrix, F, alg::ScaledDeltaAlgorithm
+    )
       @assert size(a, 1) == size(a, 2)
       @assert isdiag(a)
       @assert allequal(diagview(a))
@@ -164,13 +170,17 @@ for f in [
 end
 for f in [:eig_full!, :eig_vals!, :eigh_full!, :eigh_vals!]
   @eval begin
-    function MatrixAlgebraKit.check_input(::typeof($f), a, F, alg::DeltaAlgorithm)
+    function MatrixAlgebraKit.check_input(
+      ::typeof($f), a::AbstractMatrix, F, alg::DeltaAlgorithm
+    )
       @assert issquare(a)
       @assert isdiag(a)
       @assert all(isone, diagview(a))
       return nothing
     end
-    function MatrixAlgebraKit.check_input(::typeof($f), a, F, alg::ScaledDeltaAlgorithm)
+    function MatrixAlgebraKit.check_input(
+      ::typeof($f), a::AbstractMatrix, F, alg::ScaledDeltaAlgorithm
+    )
       @assert issquare(a)
       @assert isdiag(a)
       @assert allequal(diagview(a))
diff --git a/test/Project.toml b/test/Project.toml
index 97d324d..0f9870a 100644
--- a/test/Project.toml
+++ b/test/Project.toml
@@ -18,7 +18,7 @@ DerivableInterfaces = "0.5"
 DiagonalArrays = "0.3"
 FillArrays = "1"
 LinearAlgebra = "1"
-MatrixAlgebraKit = "0.2.5"
+MatrixAlgebraKit = "0.2.5, 0.3"
 SafeTestsets = "0.1"
 SparseArraysBase = "0.7"
 StableRNGs = "1"