Fixed typo/bug in test_notebooks.py