diff --git a/test/basis/test_type_stability.jl b/test/basis/test_type_stability.jl
index a66c4af..35296e6 100644
--- a/test/basis/test_type_stability.jl
+++ b/test/basis/test_type_stability.jl
@@ -3,16 +3,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 "Type Stability" begin
     @testset "Float64" begin