Fixed the thrust test