Fix extremely long kernel restart times

Not sure why this happens, but kernel restarts can take >30 seconds sometimes.

Probably running the notebook local and looking at the terminal will tell you more.