diff --git a/test/basis/test_partition_of_unity.jl b/test/basis/test_partition_of_unity.jl
index 2e7367e..e8b924a 100644
--- a/test/basis/test_partition_of_unity.jl
+++ b/test/basis/test_partition_of_unity.jl
@@ -2,16 +2,7 @@
 using Test
 using StaticArrays, Tensors
 
-include("../../src/topology/api.jl")
-include("../../src/topology/segments.jl")
-include("../../src/topology/triangles.jl")
-include("../../src/topology/quadrilaterals.jl")
-include("../../src/topology/tetrahedra.jl")
-include("../../src/topology/hexahedra.jl")
-include("../../src/topology/pyramids.jl")
-include("../../src/topology/wedges.jl")
-include("../../src/basis/api.jl")
-include("../../src/basis/basis_generated.jl")
+using JuliaFEM
 
 @testset "Partition of Unity" begin
     @testset "Seg2" begin