diff --git a/Model/ClassVisitor.php b/Model/ClassVisitor.php
index 3a57ad53b924d506e40852a4c9effecffdec3051..9262634dcabf037438e8f4333f02011ee1858b8c 100644
--- a/Model/ClassVisitor.php
+++ b/Model/ClassVisitor.php
@@ -8,6 +8,7 @@
 
 namespace Irstea\PlantUmlBundle\Model;
 
+use Irstea\PlantUmlBundle\Model\Decorator\NullDecorator;
 use Irstea\PlantUmlBundle\Model\Filter\AcceptAllFilter;
 use Irstea\PlantUmlBundle\Model\Namespace_\RootNamespace;
 use Irstea\PlantUmlBundle\Model\Node\Class_;
@@ -91,7 +92,7 @@ class ClassVisitor implements ClassVisitorInterface, WritableInterface
 
     /**
      *
-     * @param \Irstea\PlantUmlBundle\Model\NamespaceInterface $namespace
+     * @param NamespaceInterface $namespace
      * @param ReflectionClass $class
      * @return NodeInterface
      */
diff --git a/Model/Decorator/NullDecorator.php b/Model/Decorator/NullDecorator.php
index e55976fac5eb8177b534ac808441700f5fb5f59d..7e258a5eeddffa272d99ac4d2f1e29581704fa78 100644
--- a/Model/Decorator/NullDecorator.php
+++ b/Model/Decorator/NullDecorator.php
@@ -11,6 +11,7 @@ namespace Irstea\PlantUmlBundle\Model\Decorator;
 use Irstea\PlantUmlBundle\Model\ClassVisitorInterface;
 use Irstea\PlantUmlBundle\Model\DecoratorInterface;
 use Irstea\PlantUmlBundle\Model\NodeInterface;
+use ReflectionClass;
 
 /**
  * Description of NullDecorator
@@ -21,7 +22,7 @@ class NullDecorator implements DecoratorInterface
 {
     use \Irstea\PlantUmlBundle\Utils\Singleton;
 
-    public function decorate($className, NodeInterface $node, ClassVisitorInterface $visitor)
+    public function decorate(ReflectionClass $class, NodeInterface $node, ClassVisitorInterface $visitor)
     {
         return $this;
     }