fix 2D part of notebook

1 job for notebook_demo in 6 minutes and 33 seconds (queued for 1 second)
Status Name Job ID Coverage
  Test
passed test_min #462105

00:06:33

66.0%