Please add an assertion on the accuracy here as an non regression test. For instance assert that the score is greater than 0.95.