Hotfix: added missing while block class.