diff --git a/Project.toml b/Project.toml
index 37d7cab..eba884c 100644
--- a/Project.toml
+++ b/Project.toml
@@ -1,7 +1,7 @@
 name = "FusionTensors"
 uuid = "e16ca583-1f51-4df0-8e12-57d32947d33e"
 authors = ["ITensor developers <support@itensor.org> and contributors"]
-version = "0.5.17"
+version = "0.5.18"
 
 [deps]
 Accessors = "7d9f7c33-5ae7-4f3b-8dc6-eff91059b697"
@@ -28,7 +28,7 @@ LRUCache = "1.6"
 LinearAlgebra = "1.10"
 Random = "1.10"
 Strided = "2.3"
-TensorAlgebra = "0.6"
+TensorAlgebra = "0.6.4"
 TensorKitSectors = "0.1, 0.2"
 TypeParameterAccessors = "0.4"
 WignerSymbols = "2.0.0"
diff --git a/src/fusiontensor/tensor_algebra_interface.jl b/src/fusiontensor/tensor_algebra_interface.jl
index 2cb5399..ea64634 100644
--- a/src/fusiontensor/tensor_algebra_interface.jl
+++ b/src/fusiontensor/tensor_algebra_interface.jl
@@ -117,13 +117,16 @@ const MATRIX_FUNCTIONS = [
 
 for f in MATRIX_FUNCTIONS
     @eval begin
-        function TensorAlgebra.$f(a::FusionTensor, length_codomain::Val; kwargs...)
-            a_mat = matricize(a, length_codomain)
+        function TensorAlgebra.$f(
+                style::FusionTensorFusionStyle, a::AbstractArray,
+                length_codomain::Val; kwargs...,
+            )
+            a_mat = matricize(style, a, length_codomain)
             biperm = trivialbiperm(length_codomain, Val(ndims(a)))
             permuted_axes = axes(a)[biperm]
             checkspaces_dual(codomain(permuted_axes), domain(permuted_axes))
             fa_mat = set_data_matrix(a_mat, Base.$f(data_matrix(a_mat); kwargs...))
-            return unmatricize(fa_mat, permuted_axes)
+            return unmatricize(style, fa_mat, permuted_axes)
         end
     end
 end