diff --git a/lib/Doctrine/ORM/Tools/EntityGenerator.php b/lib/Doctrine/ORM/Tools/EntityGenerator.php
index 2946935b3..6aad04fb6 100644
--- a/lib/Doctrine/ORM/Tools/EntityGenerator.php
+++ b/lib/Doctrine/ORM/Tools/EntityGenerator.php
@@ -713,9 +713,11 @@ public function __construct()
         if (class_exists($metadata->name)) {
             $reflClass = new \ReflectionClass($metadata->name);
 
-            foreach ($reflClass->getTraits() as $trait) {
-                if ($trait->hasProperty($property)) {
-                    return true;
+            if (method_exists($reflClass, 'getTraits')) {
+                foreach ($reflClass->getTraits() as $trait) {
+                    if ($trait->hasProperty($property)) {
+                        return true;
+                    }
                 }
             }
         }
@@ -747,9 +749,11 @@ public function __construct()
         if (class_exists($metadata->name)) {
             $reflClass = new \ReflectionClass($metadata->name);
 
-            foreach ($reflClass->getTraits() as $trait) {
-                if ($trait->hasMethod($method)) {
-                    return true;
+            if (method_exists($reflClass, 'getTraits')) {
+                foreach ($reflClass->getTraits() as $trait) {
+                    if ($trait->hasMethod($method)) {
+                        return true;
+                    }
                 }
             }
         }