Fixed wrong import in Trajectory class