try fix in nb.fit