fix predict proba?