some small fixes for the Demo notebook