Fix sampling test.