Fix for regeneration task