This is the same change as in !782 (merged), but I missed doing it to the pre notebook in that MR.
@manettim @ahmedk @schmidtp