diff --git a/test/basis/test_gradient_properties.jl b/test/basis/test_gradient_properties.jl
index 8289cc1..d9d6383 100644
--- a/test/basis/test_gradient_properties.jl
+++ b/test/basis/test_gradient_properties.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 "Gradient Properties" begin
     @testset "Linear elements - constant gradients" begin