Fixing gridsearch