diff --git a/src/Generator/FileGenerator.php b/src/Generator/FileGenerator.php
index 73af4cab..c8297734 100644
--- a/src/Generator/FileGenerator.php
+++ b/src/Generator/FileGenerator.php
@@ -337,7 +337,7 @@ public function setClasses(array $classes)
 
     /**
      * @param  string $name
-     * @return ClassGenerator
+     * @return ClassGenerator|false
      */
     public function getClass($name = null)
     {