diff --git a/src/datastructures/Simulation.jl b/src/datastructures/Simulation.jl
index 2e8044a..4aa86e5 100644
--- a/src/datastructures/Simulation.jl
+++ b/src/datastructures/Simulation.jl
@@ -140,7 +140,7 @@ end
 #=
     Simulation(
         pdb_file::String,
-        atoms::AbstractVector{PDBTools.Atom},
+        atoms::AbstractVector{AtomType},
         trajectory::Chemfiles.Trajectory, frame_range::AbstractRange)
     )