diff --git a/add_attributes.function.php b/add_attributes.function.php
index 9e95113..f0c694f 100644
--- a/add_attributes.function.php
+++ b/add_attributes.function.php
@@ -6,7 +6,7 @@
 
 use Drupal\Core\Template\Attribute;
 
-$function = new Twig_SimpleFunction('add_attributes', function ($context, $additional_attributes = []) {
+$function = new \Twig\TwigFunction('add_attributes', function ($context, $additional_attributes = []) {
   if (class_exists('Drupal')) {
     $attributes = new Attribute();