Fixed small error in notebook.