Moved nbformat import into load_extension