Fix buggy notebooks