Test corrected for test_notebooks