diff --git a/Model/Model.php b/Model/Model.php
index 354439373..073436ccc 100644
--- a/Model/Model.php
+++ b/Model/Model.php
@@ -12,7 +12,6 @@
 namespace Nelmio\ApiDocBundle\Model;
 
 use Symfony\Component\PropertyInfo\Type;
-use Symfony\Component\Serializer\Normalizer\AbstractNormalizer;
 
 final class Model
 {
@@ -30,7 +29,7 @@ public function __construct(Type $type, array $groups = null, array $options = n
         $this->options = $options;
         $this->serializationContext = $serializationContext;
         if (null !== $groups) {
-            $this->serializationContext[AbstractNormalizer::GROUPS] = $groups;
+            $this->serializationContext['groups'] = $groups;
         }
     }
 
@@ -47,7 +46,7 @@ public function getType()
      */
     public function getGroups()
     {
-        return $this->serializationContext[AbstractNormalizer::GROUPS] ?? null;
+        return $this->serializationContext['groups'] ?? null;
     }
 
     /**