15c15
< index 418d2c6..02edc51 100644
---
> index 418d2c6..129f9a0 100644
58c58,70
< @@ -269,6 +278,11 @@ class PanelsIPEBlockPluginForm extends FormBase {
---
> @@ -263,12 +272,23 @@ class PanelsIPEBlockPluginForm extends FormBase {
>        return $form;
>      }
>  
> +    // Reload the panels display. If this is not done, the contexts are not
> +    // correctly populated after ajax updates. Not sure why this is.
> +    $panels_config = $this->panelsDisplay->getConfiguration();
> +    $panels_storage = \Drupal::service('panels.storage_manager');
> +    $this->panelsDisplay = $panels_storage->load($panels_config['storage_type'], $panels_config['storage_id']);
> +
>      // If a temporary configuration for this variant exists, use it.
>      $temp_store_key = $this->panelsDisplay->getTempStoreId();
>      if ($variant_config = $this->tempStore->get($temp_store_key)) {
70c82
< @@ -336,6 +350,11 @@ class PanelsIPEBlockPluginForm extends FormBase {
---
> @@ -336,6 +356,11 @@ class PanelsIPEBlockPluginForm extends FormBase {
82c94
< @@ -397,4 +416,49 @@ class PanelsIPEBlockPluginForm extends FormBase {
---
> @@ -397,4 +422,49 @@ class PanelsIPEBlockPluginForm extends FormBase {