Fix grid search test.