I have removed below code from the widget code. I'm not sure why this code is being used and what this could will break, but at the moment it did fix our problem.
// Reset storage ID if ajax was triggered. if ($form_state->getTriggeringElement()) { $storage_id = ''; }