diff --git a/.github/workflows/IntegrationTest.yml b/.github/workflows/IntegrationTest.yml
index 54dd4b0..08ceaf4 100644
--- a/.github/workflows/IntegrationTest.yml
+++ b/.github/workflows/IntegrationTest.yml
@@ -27,5 +27,6 @@ jobs:
       pkgs: |
         [
               "ITensorNetworks",
-              "ITensorNetworksNext"
+              "ITensorNetworksNext",
+              "https://github.com/ITensor/Tennis.jl"
             ]
diff --git a/Project.toml b/Project.toml
index 4f4bc03..bb78962 100644
--- a/Project.toml
+++ b/Project.toml
@@ -1,6 +1,6 @@
 name = "DataGraphs"
 uuid = "b5a273c3-7e6c-41f6-98bd-8d7f1525a36a"
-version = "0.4.0"
+version = "0.4.1"
 authors = ["Matthew Fishman <mfishman@flatironinstitute.org> and contributors"]
 
 [workspace]
@@ -14,14 +14,17 @@ SimpleTraits = "699a6c99-e7fa-54fc-8d76-47d257e15c1d"
 
 [weakdeps]
 GraphsFlows = "06909019-6f44-4949-96fc-b9d9aaa02889"
+ITensorVisualizationBase = "cd2553d2-8bef-4d93-8a38-c62f17d5ad23"
 
 [extensions]
 DataGraphsGraphsFlowsExt = "GraphsFlows"
+DataGraphsITensorVisualizationBaseExt = "ITensorVisualizationBase"
 
 [compat]
 Dictionaries = "0.4"
 Graphs = "1"
 GraphsFlows = "0.1.1"
-NamedGraphs = "0.11"
+ITensorVisualizationBase = "0.1"
+NamedGraphs = "0.11.1"
 SimpleTraits = "0.9"
-julia = "1.7"
+julia = "1.10"
diff --git a/ext/DataGraphsITensorVisualizationBaseExt/DataGraphsITensorVisualizationBaseExt.jl b/ext/DataGraphsITensorVisualizationBaseExt/DataGraphsITensorVisualizationBaseExt.jl
new file mode 100644
index 0000000..c2f9cbe
--- /dev/null
+++ b/ext/DataGraphsITensorVisualizationBaseExt/DataGraphsITensorVisualizationBaseExt.jl
@@ -0,0 +1,12 @@
+module DataGraphsITensorVisualizationBaseExt
+
+using DataGraphs: AbstractDataGraph, underlying_graph
+using ITensorVisualizationBase: ITensorVisualizationBase
+
+function ITensorVisualizationBase.visualize(graph::AbstractDataGraph, args...; kwargs...)
+    return ITensorVisualizationBase.visualize(
+        underlying_graph(graph), args...; kwargs...
+    )
+end
+
+end
diff --git a/test/Project.toml b/test/Project.toml
index 5ebc3e8..e72cd03 100644
--- a/test/Project.toml
+++ b/test/Project.toml
@@ -5,6 +5,7 @@ Dictionaries = "85a47980-9c8c-11e8-2b9f-f7ca1fa99fb4"
 Graphs = "86223c79-3864-5bf0-83f7-82e725a168b6"
 GraphsFlows = "06909019-6f44-4949-96fc-b9d9aaa02889"
 ITensorPkgSkeleton = "3d388ab1-018a-49f4-ae50-18094d5f71ea"
+ITensorVisualizationBase = "cd2553d2-8bef-4d93-8a38-c62f17d5ad23"
 NamedGraphs = "678767b0-92e7-4007-89e4-4527a8725b19"
 SafeTestsets = "1bc83da4-3b8d-516f-aca4-4fe02f6d838f"
 Suppressor = "fd094767-a336-5f1f-9728-57cf17d0bbfb"
diff --git a/test/test_itensorvisualizationbase_ext.jl b/test/test_itensorvisualizationbase_ext.jl
new file mode 100644
index 0000000..1741d42
--- /dev/null
+++ b/test/test_itensorvisualizationbase_ext.jl
@@ -0,0 +1,12 @@
+@eval module $(gensym())
+using DataGraphs: AbstractDataGraph, DataGraph
+using ITensorVisualizationBase: ITensorVisualizationBase
+using NamedGraphs.NamedGraphGenerators: named_grid
+using Test: @test, @testset
+
+@testset "DataGraphsITensorVisualizationBaseExt" begin
+    g = DataGraph(named_grid((2, 2)))
+    @test hasmethod(ITensorVisualizationBase.visualize, Tuple{AbstractDataGraph})
+    @test isnothing(ITensorVisualizationBase.visualize(g))
+end
+end