diff --git a/lib/Doctrine/ORM/Tools/EntityGenerator.php b/lib/Doctrine/ORM/Tools/EntityGenerator.php
index 948a4e689..1c55eb1bc 100644
--- a/lib/Doctrine/ORM/Tools/EntityGenerator.php
+++ b/lib/Doctrine/ORM/Tools/EntityGenerator.php
@@ -759,12 +759,12 @@ public function __construct()
      */
     protected function getTraits(ClassMetadataInfo $metadata)
     {
-        if ($metadata->reflClass != null || class_exists($metadata->name)) {
-            $reflClass = $metadata->reflClass == null
+        if ($metadata->reflClass !== null || class_exists($metadata->name)) {
+            $reflClass = $metadata->reflClass === null
                 ? new \ReflectionClass($metadata->name)
                 : $metadata->reflClass;
 
-            if (method_exists($reflClass, 'getTraits')) {
+            if (PHP_VERSION_ID >= 50400) {
                 return $reflClass->getTraits();
             }
         }
diff --git a/tests/Doctrine/Tests/ORM/Tools/EntityGeneratorTest.php b/tests/Doctrine/Tests/ORM/Tools/EntityGeneratorTest.php
index 694918d13..cd6430a2a 100644
--- a/tests/Doctrine/Tests/ORM/Tools/EntityGeneratorTest.php
+++ b/tests/Doctrine/Tests/ORM/Tools/EntityGeneratorTest.php
@@ -458,7 +458,7 @@ class EntityGeneratorTest extends \Doctrine\Tests\OrmTestCase
      */
     public function testTraitPropertiesAndMethodsAreNotDuplicated()
     {
-        if (function_exists('trait_exists')) {
+        if (PHP_VERSION_ID >= 50400) {
             $cmf = new ClassMetadataFactory();
             $em = $this->_getTestEntityManager();
             $cmf->setEntityManager($em);