diff --git a/test/basis/test_all_elements.jl b/test/basis/test_all_elements.jl
index ebf9940..e49aa93 100644
--- a/test/basis/test_all_elements.jl
+++ b/test/basis/test_all_elements.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 "All Element Types - Partition of Unity" begin
     # 1D elements