Minor improvement in simple sample test