more fixes to run the notebook