fix shape of thrust vector