Add check for trajectory