More hacks to get working on python notebook.