fix cpu tests