Fix up the tests for notebooks.