Some callback magic to get the IPynb working