diff --git a/src/PEPSKit.jl b/src/PEPSKit.jl
index 2e0142d1..2bb905b6 100644
--- a/src/PEPSKit.jl
+++ b/src/PEPSKit.jl
@@ -23,8 +23,8 @@ include("utility/hook_pullback.jl")
 include("utility/autoopt.jl")
 
 include("networks/tensors.jl")
+include("networks/infinitesquarenetwork.jl")
 
-include("states/infinitesquarenetwork.jl")
 include("states/infinitepeps.jl")
 include("states/infiniteweightpeps.jl")
 include("states/infinitepartitionfunction.jl")
diff --git a/src/states/infinitesquarenetwork.jl b/src/networks/infinitesquarenetwork.jl
similarity index 100%
rename from src/states/infinitesquarenetwork.jl
rename to src/networks/infinitesquarenetwork.jl